Better defaults

notadamking · Jul 10, 2019 · b1ca6f9 · b1ca6f9
1 parent 97427fa
commit b1ca6f9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/cli.py b/cli.py
@@ -13,7 +13,7 @@
 args = trader_cli.get_args()
 
 rewards = {"incremental-profit": IncrementalProfit, "weighted-unrealized-profit": WeightedUnrealizedProfit}
-reward_strategy = rewards[args.reward_strategy]
+reward_strategy = rewards[args.reward_strat]
 
 
 def run_optimize(args, logger):

diff --git a/lib/cli/RLTraderCLI.py b/lib/cli/RLTraderCLI.py
@@ -24,7 +24,7 @@ def __init__(self):
 
         self.parser.add_argument("--data-provider", "-d", type=str, default="static")
         self.parser.add_argument("--input-data-path", "-n", type=str, default="data/input/coinbase-1h-btc-usd.csv")
-        self.parser.add_argument("--reward-strategy", "-r", type=str, default="incremental-profit")
+        self.parser.add_argument("--reward-strategy", "-r", type=str, default="incremental-profit", dest="reward_strat")
         self.parser.add_argument("--pair", "-p", type=str, default="BTC/USD")
         self.parser.add_argument("--debug", "-D", action='store_false')
         self.parser.add_argument('--mini-batches', type=int, default=1, help='Mini batches', dest='n_minibatches')

diff --git a/optimize.py b/optimize.py
@@ -23,4 +23,4 @@ def optimize_code(params):
     from lib.RLTrader import RLTrader
 
     trader = RLTrader(**params)
-    trader.train(test_trained_model=True, render_trained_model=True)
+    trader.train(test_trained_model=True, render_test_env=True, render_report=True)