Taskο
- class dgl.graphbolt.Task[source]ο
Bases:
object
An abstract task which consists of meta information and Train/Validation/Test Set.
- meta information
The meta information of a task includes any kinds of data that are defined by the user in YAML when instantiating the task.
- Train/Validation/Test Set
The train/validation/test (TVT) set which is used to train the neural networks. We calculate the embeddings based on their respective features and the graph structure, and then utilize the embeddings to optimize the neural network parameters.
- property test_set: ItemSet | ItemSetDictο
Return the test set.
- property train_set: ItemSet | ItemSetDictο
Return the training set.
- property validation_set: ItemSet | ItemSetDictο
Return the validation set.