From cb93af3155c318b9168fdec272d5203a57de3d47 Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Wed, 7 Nov 2018 15:09:15 +0000 Subject: Replace hardcoded constant --- train.py | 4 ++-- 1 file 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])) -- cgit v1.2.3-54-g00ecf