update

1 files changed, 11 insertions, 2 deletions

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