.

SamuelDiai · Apr 1, 2021 · 8dcefce · 8dcefce
1 parent d0b0f54
commit 8dcefce
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/animals.py b/animals.py
@@ -37,4 +37,4 @@ def animals(k, n_iter, alpha, beta):
     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')
     fig.savefig(filename)
-    return G
+    return graph
diff --git a/basic_experiments.py b/basic_experiments.py
@@ -65,7 +65,7 @@ def load_dataset_and_sgl(dataset, k, k_sgl, n):
     plt.ylabel('y-coordinate')
     filename = os.path.join(plots_dir, dataset, 'graph_%s_%s_%s' % (k , k_sgl, n))
     fig.savefig(filename)
-    return G
+    return graph
 
 
 def two_moons(n, k_sgl):

diff --git a/cancer.py b/cancer.py
@@ -37,4 +37,4 @@ def Cancer(df_cancer, y_cancer, alpha, beta, k, n_iter):
     plt.title("Learned graph for the cancer dataset k=%s n_iter=%s alpha=%.3f beta=%.3f" % (k , n_iter, alpha, beta))
     filename = os.path.join(plots_dir, 'cancer', 'graph')
     fig.savefig(filename)
-    return G
+    return graph