任务

class dgl.graphbolt.Task[source]

Bases: object

一个抽象的任务，包含元信息以及训练/验证/测试集。

  • 元信息

    任务的元信息包括用户在通过 YAML 实例化任务时定义的任何类型的数据。

  • 训练/验证/测试集

    训练/验证/测试 (TVT) 集用于训练神经网络。我们根据它们各自的特征和图结构计算嵌入，然后利用这些嵌入来优化神经网络参数。

property metadata: Dict

返回任务元信息。

property test_set: ItemSet | HeteroItemSet

返回测试集。

property train_set: ItemSet | HeteroItemSet

返回训练集。

property validation_set: ItemSet | HeteroItemSet

返回验证集。