diff options
Diffstat (limited to 'stock.py')
-rwxr-xr-x | stock.py | 62 |
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() |