aboutsummaryrefslogtreecommitdiffstats
path: root/seer.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rwxr-xr-xseer.py13
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()