Actor数据集

class dgl.data.ActorDataset(raw_dir=None, force_reload=False, verbose=True, transform=None)

基础类：DGLBuiltinDataset

电影-导演-演员-编剧网络中仅包含演员的诱导子图 来自大规模网络中的社会影响分析 ，由 Geom-GCN: 几何图卷积网络 引入

节点代表演员，边代表在同一维基百科页面上的共现。节点特征对应于维基百科页面中的一些关键词。

统计：

• 节点数：7600

• 边数: 33391

• 班级数量：5

• 10 个训练/验证/测试分割

  • 训练: 3648

  • 值: 2432

  • 测试: 1520

Parameters:

• raw_dir (str, optional) – 用于存储处理后的数据的原始文件目录。默认值：~/.dgl/

• force_reload (bool, 可选) – 是否重新下载数据源。默认值：False

• verbose (bool, optional) – 是否打印进度信息。默认值：True

• transform (callable, optional) – 一个转换函数，它接收一个 DGLGraph 对象并返回 一个转换后的版本。每次访问前，DGLGraph 对象都会被 转换。默认值：None

num_classes

节点类的数量

Type:

int

注释

该图不包含双向的边。

__getitem__(idx)

获取索引处的数据对象。

__len__()

数据集中的示例数量。