提交 14b2abba 编写于 作者: Z ZHUI

delete some comments

上级 6bb55f73
...@@ -31,18 +31,6 @@ from collections import namedtuple ...@@ -31,18 +31,6 @@ from collections import namedtuple
import pgl import pgl
import numpy as np import numpy as np
import copy import copy
"""
dict_keys(['edge_index', 'edge_feat', 'node_feat', 'node_year', 'num_nodes'])
edge_index shape: (2, 1166243)
edge_index type: <class 'numpy.ndarray'>
[[104447 15858 107156 ... 45118 45118 45118]
[ 13091 47283 69161 ... 162473 162537 72717]]
edge_feat: None
node_feat shape: (169343, 128)
node_year shape: (169343, 1)
num_nodes: 169343
label shape: (169343, 1)
"""
def traverse(item): def traverse(item):
...@@ -131,7 +119,6 @@ class ArxivDataGenerator(BaseDataGenerator): ...@@ -131,7 +119,6 @@ class ArxivDataGenerator(BaseDataGenerator):
labels = label[nodes_idx] labels = label[nodes_idx]
self.nodes_idx = nodes_idx self.nodes_idx = nodes_idx
self.labels = labels self.labels = labels
#self.static_gw_based_line_example(nodes_idx, labels)
self.sample_based_line_example(nodes_idx, labels) self.sample_based_line_example(nodes_idx, labels)
def sample_based_line_example(self, nodes_idx, labels): def sample_based_line_example(self, nodes_idx, labels):
...@@ -142,11 +129,6 @@ class ArxivDataGenerator(BaseDataGenerator): ...@@ -142,11 +129,6 @@ class ArxivDataGenerator(BaseDataGenerator):
print("Phase", self.phase) print("Phase", self.phase)
print("Len Examples", len(self.line_examples)) print("Len Examples", len(self.line_examples))
def batch_fn2(self, batch_ex):
feed_dict["batch_nodes"] = np.array(batch_ex[0]['node'], dtype="int64")
feed_dict["labels"] = np.array(batch_ex[0]['label'], dtype="int64")
return feed_dict
def batch_fn(self, batch_ex): def batch_fn(self, batch_ex):
batch_nodes = [] batch_nodes = []
cc = 0 cc = 0
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册