fixed networkx issue

Author: Chenhui1016

graphzoom/scoring.py

@@ -18,8 +18,8 @@ def lr(dataset_dir, data_dir, dataset):
     G      = json_graph.node_link_graph(json.load(open(dataset_dir + "/{}-G.json".format(dataset))))
     labels = json.load(open(dataset_dir + "/{}-class_map.json".format(dataset)))
 
-    train_ids    = [n for n in G.nodes() if not G.node[n]['val'] and not G.node[n]['test']]
-    test_ids     = [n for n in G.nodes() if G.node[n]['test']]
+    train_ids    = [n for n in G.nodes() if not G.nodes[n]['val'] and not G.nodes[n]['test']]
+    test_ids     = [n for n in G.nodes() if G.nodes[n]['test']]
     test_ids     = test_ids[:1000]
     train_labels = [labels[str(i)] for i in train_ids]
     test_labels  = [labels[str(i)] for i in test_ids]