# Make shell script option parser
parser = argparse.ArgumentParser()
parser.add_argument("--train",
help="Run model training loops and save models \
in models/ directory.",
action="store_true")
parser.add_argument("--test",
help="Run testing for all models saved in \
models/ directory.",
action="store_true")
args = parser.parse_args()
# Get datasets
pp = PreProcessor(data_dir="../data/")
train, val, test = pp.get_data()
if args.train:
# Prepare models for training.
timesteps = [2, 5, 10, 20, 50, 100, 200]
models = [LinearModel, Layer1NN, Layer2NN, Layer3NN]
_trainer = Trainer(timesteps, models, train, val)
_trainer.train_loop()
elif args.test:
del train
del val
models_dir = list(
