small bug transpose of phi in plotting

adbrebs · Mar 31, 2016 · 2eb7432 · 2eb7432
1 parent 89c1bdd
commit 2eb7432
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/extensions.py b/extensions.py
@@ -78,7 +78,8 @@ def execute_virtual(self, batch_id):
         # plot_coord(coord_gen,
         #            folder_path=self.folder_path,
         #            file_name='{}_'.format(batch_id) + self.file_name)
-        mats = [(a_gen, 'alpha'), (k_gen, 'kapa'), (p_gen.T, 'phi'),
+        p_gen = np.swapaxes(p_gen, 1, 2)
+        mats = [(a_gen, 'alpha'), (k_gen, 'kapa'), (p_gen, 'phi'),
                 (w_gen, 'omega')]
         plot_generated_sequences(
             coord_gen, mats,

diff --git a/main_cond.py b/main_cond.py
@@ -31,7 +31,7 @@
 gain = 0.01
 batch_size = 50  # batch_size
 chunk = None
-every = 1
+every = 100
 every_val = 1000
 sample_strings = ['Jose is a raccoon !']*4
 tmp_path = os.environ.get('TMP_PATH')

diff --git a/sample_model.py b/sample_model.py
@@ -59,7 +59,8 @@
         coord_ini_mat, cond, cond_mask,
         h_ini_mat, k_ini_mat, w_ini_mat, options.bias)
     print 'done in {} seconds'.format(time.clock()-beg)
-    mats = [(a_gen, 'alpha'), (k_gen, 'kapa'), (p_gen.T, 'phi'),
+    p_gen = np.swapaxes(p_gen, 1, 2)
+    mats = [(a_gen, 'alpha'), (k_gen, 'kapa'), (p_gen, 'phi'),
             (w_gen, 'omega')]
     print 'Printing...',
     beg = time.clock()