瀏覽代碼

Autoencoder encoder graph fix

Min 4 年之前
父節點
當前提交
328596b16f
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      models/autoencoder.py

+ 1 - 1
models/autoencoder.py

@@ -240,7 +240,7 @@ class Autoencoder(Model):
 
 
 def view_encoder(encoder, N, samples=1000, title="Autoencoder generated alphabet"):
-    test_values = misc.generate_random_bit_array(samples).reshape((-1, N))
+    test_values = misc.generate_random_bit_array(samples*N).reshape((-1, N))
     test_values_ho = misc.bit_matrix2one_hot(test_values)
     mvector = np.array([2 ** i for i in range(N)], dtype=int)
     symbols = (test_values * mvector).sum(axis=1)