diff options
| -rwxr-xr-x | train.py | 3 |
1 files changed, 2 insertions, 1 deletions
@@ -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') |