|
@@ -12,8 +12,8 @@ from lab import generator, trainer, benchmark
|
|
|
|
|
|
# Parse script arguments
|
|
|
parser = argparse.ArgumentParser(description = "Start a training session for a neuronal network and display results.")
|
|
|
-parser.add_argument('--alpha', help = "set the learning rate", dest="learnRate", type = float, default = 0.05)
|
|
|
-parser.add_argument('--epochs', help = "set the number of iterations", dest="epochs", type = int, default = 2)
|
|
|
+parser.add_argument('-a, --alpha', help = "set the learning rate", dest="learnRate", type = float, default = 0.05)
|
|
|
+parser.add_argument('-e, --epochs', help = "set the number of iterations", dest="epochs", type = int, default = 2)
|
|
|
|
|
|
args = parser.parse_args()
|
|
|
|