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 metadata: Dict

Return the task metadata.

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.