Replace hardcoded constant

1 files changed, 2 insertions, 2 deletions

diff --git a/train.py b/train.py
index 917c895..6ac963b 100755
--- a/train.py
+++ b/train.py
@@ -213,8 +213,8 @@ def main():
     faces_train, faces_test, target_train, target_test = test_split(n_faces, raw_faces, args.split, args.seed)
 
     if args.classifyalt:
-        faces_train = faces_train.reshape(n_faces, 8, n_pixels)
-        target_train = target_train.reshape(n_faces, 8)
+        faces_train = faces_train.reshape(n_faces, int(faces_train.shape[0]/n_faces), n_pixels)
+        target_train = target_train.reshape(n_faces, int(target_train.shape[0]/n_faces))
 
         accuracy = np.zeros(n_faces)
         distances = np.zeros((n_faces, faces_test.shape[0]))