diff options
Diffstat (limited to '')
| -rwxr-xr-x | seer.py | 13 |
1 files changed, 11 insertions, 2 deletions
@@ -7,7 +7,7 @@ import readline import signal import sys from ltsm import lstm_type_1, lstm_type_2, lstm_type_3 -from cnn import cnn_type_1 +from cnn import marrionette_type_1 def SigHandler_SIGINT(signum, frame): print() @@ -16,15 +16,24 @@ def SigHandler_SIGINT(signum, frame): class Argparser(object): def __init__(self): parser = argparse.ArgumentParser() - parser.add_argument("--string", type=str, help="string") + parser.add_argument("--which", type=str, help="which one to run") parser.add_argument("--bool", action="store_true", help="bool", default=False) parser.add_argument("--dbg", action="store_true", help="debug", default=False) self.args = parser.parse_args() +def launch_ais(which): + if which == "marionette": marrionette_type_1() + elif which == "lstm_type_1": lstm_type_1() + elif which == "lstm_type_2": lstm_type_2() + elif which == "lstm_type_3": lstm_type_3() + elif which == "cnn_type_1": pass + else: pass + # write code here def premain(argparser): signal.signal(signal.SIGINT, SigHandler_SIGINT) #here + launch_ais(argparser.args.which) def main(): argparser = Argparser() |