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ΒΆ

Return the task metadata.

property test_setΒΆ

Return the test set.

property train_setΒΆ

Return the training set.

property validation_setΒΆ

Return the validation set.