1 files changed, 2 insertions, 2 deletions
diff --git a/train.py b/train.py
index b7e9fb0..a0c2a9a 100755
--- a/train.py
+++ b/train.py
@@ -18,13 +18,13 @@ def normalise_faces(average_face, raw_faces):
 parser = argparse.ArgumentParser()
 parser.add_argument("-i", "--data", help="Input CSV file", required=True)
 parser.add_argument("-o", "--model", help="Output model file", required=True)
-parser.add_argument("-m", "--M", help="Number of eigenvalues in model", type=int)
+parser.add_argument("-m", "--eigen", help="Number of eigenvalues in model", required=True, type=int)
 args = parser.parse_args()
 
 assert args.data, "No input CSV data (-i, --input-data)"
 assert args.model, "No model specified (-o, --model)"
 
-M = args.M | -1;
+M = args.eigen
 
 raw_faces = genfromtxt(args.data, delimiter=',').T