mindspore_gl.dataset.IMDBBinary

class mindspore_gl.dataset.IMDBBinary(root)

IMDBBinary数据集，用于读取和解析IMDBBinary数据集的源数据集。

关于IMDBBinary数据集：

IMDBBinary数据集，用于读取和解析IMDBBinaary数据集的源数据集。IMDB-BINARY是一个电影协作数据集，由1000名在IMDB电影中扮演角色的演员组成的角色扮演网络组成。 在每个图中，节点表示演员，如果他们出现在同一部电影中，则它们之间有一条边。这些图来源于动作片和浪漫片。

信息统计：

• 节点: 19773

• 边: 193062

• 图： 1000

• 分类数量: 2

• 数据集切分:

  • Train: 800

  • Valid: 200

下载地址：https://ls11-www.cs.tu-dortmund.de/people/morris/graphkerneldatasets/IMDB-BINARY.zip 。 您可以将数据集文件组织到以下目录结构中进行读取。

.
├── IMDB-BINARY_A.txt
├── IMDB-BINARY_graph_indicator.txt
└── IMDB-BINARY_graph_labels.txt

参数：

• root (str) - 包含imdb_binary_with_mask.npz的根目录的路径。

异常：

• TypeError - 如果 root 不是str。

• RuntimeError - 如果 root 不包含数据文件。

样例：

>>> from mindspore_gl.dataset.imdb_binary import IMDBBinary
>>> root = "path/to/imdb_binary"
>>> dataset = IMDBBinary(root)

property edge_feat_size

标签类数量。

返回：

int，类的数量。

样例：

>>> #dataset is an instance object of Dataset
>>> edge_feat_size = dataset.edge_feat_size

property graph_count

图的总数。

返回：

int，图的数量。

样例：

>>> #dataset is an instance object of Dataset
>>> graph_count = dataset.graph_count

property graph_edges

累计图边数。

返回：

numpy.ndarray，累积边数组。

样例：

>>> #dataset is an instance object of Dataset
>>> val_mask = dataset.graph_edges

property graph_label

图的标签。

返回：

numpy.ndarray，图标签数组。

样例：

>>> #dataset is an instance object of Dataset
>>> graph_label = dataset.graph_label

graph_node_feat(graph_idx)

图上每个节点的特征。

参数：

• graph_idx (int) - 图索引。

返回：

numpy.ndarray，图的节点特征。

样例：

>>> #dataset is an instance object of Dataset
>>> graph_node_feat = dataset.graph_node_feat(graph_idx)

property graph_nodes

累计图节点数。

返回：

numpy.ndarray，累计节点数组。

样例：

>>> #dataset is an instance object of Dataset
>>> val_mask = dataset.graph_nodes

property node_feat

节点特征。

返回：

numpy.ndarray，节点特征数组。

样例：

>>> #dataset is an instance object of Dataset
>>> node_feat = dataset.node_feat

property node_feat_size

每个节点的特征数量。

返回：

int，特征的数量。

样例：

>>> #dataset is an instance object of Dataset
>>> node_feat_size = dataset.node_feat_size

property num_classes

图标签种类。

返回：

int，图标签的种类。

样例：

>>> #dataset is an instance object of Dataset
>>> num_classes = dataset.num_classes

property train_graphs

训练图ID。

返回：

numpy.ndarray，训练图ID。

样例：

>>> #dataset is an instance object of Dataset
>>> train_graphs = dataset.train_graphs

property train_mask

训练节点掩码。

返回：

numpy.ndarray，掩码数组。

样例：

>>> #dataset is an instance object of Dataset
>>> train_mask = dataset.train_mask

property val_graphs

校验的图ID。

返回：

numpy.ndarray，校验图ID数组。

样例：

>>> #dataset is an instance object of Dataset
>>> val_graphs = dataset.val_graphs

property val_mask

校验节点掩码。

返回：

numpy.ndarray，掩码数组。

样例：

>>> #dataset is an instance object of Dataset
>>> val_mask = dataset.val_mask