Change number of n_cases to correct one

1 files changed, 1 insertions, 1 deletions

diff --git a/train.py b/train.py
index fd03220..aed8f6e 100755
--- a/train.py
+++ b/train.py
@@ -29,7 +29,7 @@ def normalise_faces(average_face, faces):
 # Split data into training and testing sets
 def test_split(n_faces, raw_faces, split, seed):
     random.seed(seed)
-    n_cases = 52
+    n_cases = 10
     n_pixels = 2576 
     
     raw_faces_split = np.split(raw_faces,n_cases)