From 3b283614e35fc04214c4e7d8bc7acd07037520ac Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Thu, 18 Oct 2018 18:34:25 +0100 Subject: Add neighbor argument --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'train.py') 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') -- cgit v1.2.3-54-g00ecf