From 94cc1d159e62c5dce5d43faede3e85f63f451cdd Mon Sep 17 00:00:00 2001 From: bloodstalker Date: Sun, 2 Sep 2018 16:42:56 +0430 Subject: update --- seer.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) (limited to 'seer.py') diff --git a/seer.py b/seer.py index c63c6cf..e95a6d8 100755 --- a/seer.py +++ b/seer.py @@ -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() -- cgit v1.2.3