Sfoglia il codice sorgente

Add shortcuts for arguments

DricomDragon 5 anni fa
parent
commit
26d3d2c500
1 ha cambiato i file con 2 aggiunte e 2 eliminazioni
  1. 2 2
      python/start_session.py

+ 2 - 2
python/start_session.py

@@ -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()