Explorar o código

Autoencoder encoder graph fix

Min %!s(int64=4) %!d(string=hai) anos
pai
achega
328596b16f
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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)