aboutsummaryrefslogtreecommitdiffstats
path: root/stock.py
diff options
context:
space:
mode:
Diffstat (limited to 'stock.py')
-rwxr-xr-xstock.py62
1 files changed, 62 insertions, 0 deletions
diff --git a/stock.py b/stock.py
new file mode 100755
index 0000000..6ad60ed
--- /dev/null
+++ b/stock.py
@@ -0,0 +1,62 @@
+#!python
+# _*_ coding=utf-8 _*_
+
+import argparse
+import code
+import readline
+import signal
+import sys
+import pandas as pd
+import numpy as np
+import json
+import os
+import urllib3
+import time
+from keras.models import Sequential
+from keras.layers import Activation, Dense
+from keras.layers import LSTM
+from keras.layers import Dropout
+from keras.models import load_model
+
+def SigHandler_SIGINT(signum, frame):
+ print()
+ sys.exit(0)
+
+class Argparser(object):
+ def __init__(self):
+ parser = argparse.ArgumentParser()
+ parser.add_argument("--string", type=str, help="string")
+ 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 build_model(train_data):
+ model = models.Sequential()
+ model.add(layers.Dense(64, activation="relu", input_shape=(train_data.shape[1],)))
+ model.add(layers.Dense(64, activation="relu"))
+ model.add(layers.Dense(1))
+ model.compile(optimizer="rmsprop", loss="mse", metrics=["acc"])
+ return model
+
+# write code here
+def premain(argparser):
+ signal.signal(signal.SIGINT, SigHandler_SIGINT)
+ #here
+
+def main():
+ argparser = Argparser()
+ if argparser.args.dbg:
+ try:
+ premain(argparser)
+ except Exception as e:
+ print(e.__doc__)
+ if e.message: print(e.message)
+ variables = globals().copy()
+ variables.update(locals())
+ shell = code.InteractiveConsole(variables)
+ shell.interact(banner="DEBUG REPL")
+ else:
+ premain(argparser)
+
+if __name__ == "__main__":
+ main()