def main(args):
env = Market(args.codes)
algorithm = Algorithm(env, env.trader.action_space, env.data_dim, **{
# "mode": args.mode,
# "mode": "test",
"episodes": 10,
# "log_level": args.log_level,
})
algorithm.run()
def main(args):
env = Market(args.codes)
algorithm = Algorithm(tf.Session(config=config), env, env.trader.action_space, env.data_dim, **{
"mode": args.mode,
# "mode": "test",
"log_level": args.log_level,
"save_path": os.path.join(CHECKPOINTS_DIR, "RL", "DDPG", "model"),
"enable_saver": True,
})
algorithm.run()
algorithm.eval_and_plot()
def main(args):
env = Market(args.codes, **{"use_sequence": True})
algorithm = Algorithm(tf.Session(config=config), env, env.seq_length, env.data_dim, env.code_count, **{
# "mode": args.mode,
"mode": "test",
"log_level": args.log_level,
"save_path": os.path.join(CHECKPOINTS_DIR, "SL", "DualAttnRNN", "model"),
"enable_saver": True,
})
algorithm.run()
algorithm.eval_and_plot()