Non-dividable sample number fix

models/autoencoder.py

@@ -5,6 +5,7 @@ import tensorflow as tf
 from sklearn.metrics import accuracy_score
 from tensorflow.keras import layers, losses
 from tensorflow.keras.models import Model
+from functools import partial
 import misc
 import defs
 
@@ -99,7 +100,10 @@ class Autoencoder(Model):
         x_train = misc.generate_random_bit_array(samples).reshape((-1, self.N))
         x_train_ho = misc.bit_matrix2one_hot(x_train)
 
-        x_test_array = misc.generate_random_bit_array(samples * 0.3)
+        test_samples = samples * 0.3
+        if test_samples % self.N:
+            test_samples += self.N - (test_samples % self.N)
+        x_test_array = misc.generate_random_bit_array(test_samples)
         x_test = x_test_array.reshape((-1, self.N))
         x_test_ho = misc.bit_matrix2one_hot(x_test)