Tutorial AE MNIST added added (corrections 2)

MelJan · Jan 28, 2018 · 842c493 · 842c493
1 parent b871de5
commit 842c493
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/examples/GRBM_2D.py b/examples/GRBM_2D.py
@@ -59,7 +59,8 @@
 
 # split training test data
 train_data = whitened_data[0:numx.int32(whitened_data.shape[0] / 2.0), :]
-test_data = whitened_data[numx.int32(whitened_data.shape[0] / 2.0):whitened_data.shape[0], :]
+test_data = whitened_data[numx.int32(whitened_data.shape[0] / 2.0
+                                     ):whitened_data.shape[0], :]
 
 # Input output dims
 h1 = 2
@@ -194,22 +195,28 @@
     vis.calculate_amari_distance(zca.project(mixing_matrix.T), ica.projection_matrix.T))
 
 print "Amari distanca between true mixing matrix and GRBM weight vector 1 and 2: "+str(
-    vis.calculate_amari_distance(zca.project(mixing_matrix.T), numx.vstack((rbm.w.T[0:1],rbm.w.T[1:2]))))
+    vis.calculate_amari_distance(zca.project(mixing_matrix.T),
+                                 numx.vstack((rbm.w.T[0:1],rbm.w.T[1:2]))))
 
 print "Amari distanca between true mixing matrix and GRBM weight vector 1 and 3: "+str(
-    vis.calculate_amari_distance(zca.project(mixing_matrix.T), numx.vstack((rbm.w.T[0:1],rbm.w.T[2:3]))))
+    vis.calculate_amari_distance(zca.project(mixing_matrix.T),
+                                 numx.vstack((rbm.w.T[0:1],rbm.w.T[2:3]))))
 
 print "Amari distanca between true mixing matrix and GRBM weight vector 1 and 4: "+str(
-    vis.calculate_amari_distance(zca.project(mixing_matrix.T), numx.vstack((rbm.w.T[0:1],rbm.w.T[3:4]))))
+    vis.calculate_amari_distance(zca.project(mixing_matrix.T),
+                                 numx.vstack((rbm.w.T[0:1],rbm.w.T[3:4]))))
 
 print "Amari distanca between true mixing matrix and GRBM weight vector 2 and 3: "+str(
-    vis.calculate_amari_distance(zca.project(mixing_matrix.T), numx.vstack((rbm.w.T[1:2],rbm.w.T[2:3]))))
+    vis.calculate_amari_distance(zca.project(mixing_matrix.T),
+                                 numx.vstack((rbm.w.T[1:2],rbm.w.T[2:3]))))
 
 print "Amari distanca between true mixing matrix and GRBM weight vector 2 and 4: "+str(
-    vis.calculate_amari_distance(zca.project(mixing_matrix.T), numx.vstack((rbm.w.T[1:2],rbm.w.T[3:4]))))
+    vis.calculate_amari_distance(zca.project(mixing_matrix.T),
+                                 numx.vstack((rbm.w.T[1:2],rbm.w.T[3:4]))))
 
 print "Amari distanca between true mixing matrix and GRBM weight vector 3 and 4: "+str(
-    vis.calculate_amari_distance(zca.project(mixing_matrix.T), numx.vstack((rbm.w.T[2:3],rbm.w.T[3:4]))))
+    vis.calculate_amari_distance(zca.project(mixing_matrix.T),
+                                 numx.vstack((rbm.w.T[2:3],rbm.w.T[3:4]))))
 
 # Display results
 # create a new figure of size 5x5

diff --git a/examples/RBM_MNIST_small.py b/examples/RBM_MNIST_small.py
@@ -117,7 +117,7 @@
         ll_test = numx.mean(estimator.log_likelihood_v(rbm, logZ, test_data))
         re = numx.mean(estimator.reconstruction_error(rbm, train_data))
         print('{}\t\t{:.4f}\t\t\t{:.4f}\t\t\t\t{:.4f}\t\t\t{}'.format(
-            epoch+1, re, ll_train, ll_test, measurer.get_expected_end_time(epoch+1, epochs)))
+        epoch+1, re, ll_train, ll_test, measurer.get_expected_end_time(epoch+1, epochs)))
     else:
         print(epoch+1)