lyq459477549的博客

原创 NN4G代码分析

代码中提供了三种数据集，分别是NCI1,MUTAG和IMDB-MULTI，在这里以NCI1数据集举例，这是一个用于二分类的数据集首先，数据集中包含四个文件，可以将每一个分子视为一个图，分子中的原子是图的节点，原子之间的键对应图的边。这个数据集中共包含了4110个分子（图）和122747个原子（节点）。并对图和节点分别进行统一的编号分别代表了A：每条边对应的节点，格式如下graph_indicator：节点属于哪个图graph_labels：每个图的标签node_labels：每个节点的标签。