From 14b2abbaba57a1b044ce8cc10bceec57158ae65e Mon Sep 17 00:00:00 2001 From: ZHUI Date: Tue, 16 Jun 2020 11:25:46 +0800 Subject: [PATCH] delete some comments --- .../dataloader/ogbn_arxiv_dataloader.py | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/ogb_examples/nodeproppred/ogbn-arxiv/dataloader/ogbn_arxiv_dataloader.py b/ogb_examples/nodeproppred/ogbn-arxiv/dataloader/ogbn_arxiv_dataloader.py index 891d5b9..48529c6 100644 --- a/ogb_examples/nodeproppred/ogbn-arxiv/dataloader/ogbn_arxiv_dataloader.py +++ b/ogb_examples/nodeproppred/ogbn-arxiv/dataloader/ogbn_arxiv_dataloader.py @@ -31,18 +31,6 @@ from collections import namedtuple import pgl import numpy as np import copy -""" -dict_keys(['edge_index', 'edge_feat', 'node_feat', 'node_year', 'num_nodes']) -edge_index shape: (2, 1166243) -edge_index type: -[[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): @@ -131,7 +119,6 @@ class ArxivDataGenerator(BaseDataGenerator): labels = label[nodes_idx] self.nodes_idx = nodes_idx self.labels = labels - #self.static_gw_based_line_example(nodes_idx, labels) self.sample_based_line_example(nodes_idx, labels) def sample_based_line_example(self, nodes_idx, labels): @@ -142,11 +129,6 @@ class ArxivDataGenerator(BaseDataGenerator): print("Phase", self.phase) 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): batch_nodes = [] cc = 0 -- GitLab