读K-GNN代码记录

bipartite graph  &&  binary graph

1-IMDB数据集的一些属性：

IMDB-BINARY

• Data(edge_index=[2, 188188], x=[19409, 136], y=[996])
• dataset.num_features:  136
• dataset.num_classes:  2
• dataset.num_node_features: 136
• dataset.num_node_labels: 136
• dataset.num_node_attributes:  0
• dataset.num_edge_attributes: 0
• dataset.num_edge_features: 0
• dataset.num_edge_labels: 0

X表示节点特征，shape: [num_nodes, num_node_edges]

Y表示label，shape: [num_nodes, num_label_dimension]，或者是 [graph_label_dimension]，这里的Y是图的标签

edge_index表示边，即邻接表，shape: [2, num_edges]

需要的有label, features,edge

 

name of dataset: PROTEINS

Data(edge_index=[2, 156166], x=[41859, 3], y=[1109])
lenth 1109
edge_index of data: tensor([[ 0,  0,  0,  ..., 39, 39, 39],
        [11, 22, 32,  ...,  7, 37, 38]])
x of data: tensor([[1., 0., 0.],
        [1., 0., 0.],
        [1., 0., 0.],
        ...,
        [0., 0., 1.],
        [0., 0., 1.],
        [0., 0., 1.]])
y of data: tensor([0, 0, 0,  ..., 1, 1, 1])
dataset.num_features:  3
dataset.num_classes:  2
dataset.num_node_features: 3
dataset.num_node_labels: 3
dataset.num_node_attributes:  0
dataset.num_edge_attributes: 0
dataset.num_edge_features: 0
dataset.num_edge_labels: 0
---------------- print over ----------------

PICTURE:/home/lll/.config/tencent-qq//AppData/file//sendpix1.jpg