Add neighbor argument

1 files changed, 2 insertions, 1 deletions

diff --git a/train.py b/train.py
index 134b70a..26567e8 100755
--- a/train.py
+++ b/train.py
@@ -28,7 +28,8 @@ def normalise_faces(average_face, raw_faces):
 # usage: train.py [-h] -i DATA -o MODEL [-m M]
 parser = argparse.ArgumentParser()
 parser.add_argument("-i", "--data", help="Input CSV file", required=True)
-parser.add_argument("-m", "--eigen", help="Number of eigenvalues in model", required=True, type=int)
+parser.add_argument("-m", "--eigen", help="Number of eigenvalues in model", type=int, default = 8 )
+parser.add_argument("-n", "--neighbors", help="How many neighbors to use", type=int, default = 3)
 parser.add_argument("-g", "--graph", help="Should we show graphs", action='store_true')
 parser.add_argument("-p", "--pca", help="Use PCA", action='store_true')
 parser.add_argument("-l", "--lda", help="Use LDA", action='store_true')