utils main

JiaxuanYou · Jun 6, 2018 · 3444b8a · 3444b8a
1 parent 51ef81b
commit 3444b8a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/baselines/mmsb.py b/baselines/mmsb.py
@@ -150,5 +150,5 @@ def graph_gen_from_blockmodel(B, Z):
                 gen_graphs.append(graph_gen_from_blockmodel(B, Zp))
 
     save_path = '/lfs/local/0/rexy/graph-generation/eval_results/mmsb/'
-    utils.save_graph_list(gen_graphs, os.path.join(save_path, prog_args.dataset + '1.dat'))
+    utils.save_graph_list(gen_graphs, os.path.join(save_path, prog_args.dataset + '.dat'))
 
diff --git a/utils.py b/utils.py
@@ -510,7 +510,9 @@ def test_perturbed():
 if __name__ == '__main__':
     #test_perturbed()
     #graphs = load_graph_list('graphs/' + 'GraphRNN_RNN_community4_4_128_train_0.dat')
-    graphs = load_graph_list('graphs/' + 'GraphRNN_RNN_community4_4_128_pred_2500_1.dat')
+    #graphs = load_graph_list('graphs/' + 'GraphRNN_RNN_community4_4_128_pred_2500_1.dat')
+    graphs = load_graph_list('eval_results/mmsb/' + 'community41.dat')
 
-    draw_graph_list(graphs[:16], 4, 4, fname='figures/gen_2500')
+    for i in range(0, 160, 16):
+        draw_graph_list(graphs[i:i+16], 4, 4, fname='figures/community4_' + str(i))