.

SamuelDiai · Apr 1, 2021 · a7046ef · a7046ef
1 parent 6a3dc31
commit a7046ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/animals.py b/animals.py
@@ -32,7 +32,7 @@ def animals(k, n_iter, alpha, beta):
         mapping[i] = animals_names[i, 0]
 
     G = nx.relabel_nodes(G, mapping)
-    fig = plt.figure(figsize=(15,15))
+    fig = plt.figure(figsize=(10,10))
     nx.draw(G, with_labels=True, font_weight='bold')
     plt.title("Learned graph for the animal dataset k=%s n_iter=%s alpha=%.3f beta=%.3f" % (k , n_iter, alpha, beta))
     filename = os.path.join(plots_dir, 'animals', 'graph')

diff --git a/basic_experiments.py b/basic_experiments.py
@@ -57,7 +57,7 @@ def load_dataset_and_sgl(dataset, k, k_sgl, n):
     norm_weights = norm_weights
 
     # plot graph
-    fig = plt.figure(figsize=(15,15))
+    fig = plt.figure(figsize=(10,10))
     nx.draw_networkx(G,pos, width=norm_weights)
     plt.title("Learned graph for %s dataset" % dataset)
     plt.suptitle('SGL components k_sgl=%s, Real components k=%s' % (k_sgl, k))