소스 검색

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)