Merge branch 'master' of https://github.com/quantylab/rltrader

howtokim · May 20, 2018 · 1372fc5 · 1372fc5
2 parents c1f5892 + 38de268
commit 1372fc5
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 141 deletions.
diff --git a/_main.py b/_main.py
diff --git a/agent.py b/agent.py
@@ -7,7 +7,7 @@ class Agent:
 
     # 매매 수수료 및 세금
     TRADING_CHARGE = 0  # 거래 수수료 미고려 (일반적으로 0.015%)
-    TRADING_TAX = 0  # 거래세 미고려 (실제0.3%)
+    TRADING_TAX = 0  # 거래세 미고려 (실제 0.3%)
 
     # 행동
     ACTION_BUY = 0  # 매수

diff --git a/policy_learner.py b/policy_learner.py
@@ -147,7 +147,7 @@ def fit(
 
                 # 학습 모드이고 지연 보상이 존재할 경우 정책 신경망 갱신
                 if delayed_reward == 0 and batch_size >= max_memory:
-                    delayed_reward = self.agent.immediate_reward
+                    delayed_reward = immediate_reward
                 if learning and delayed_reward != 0:
                     # 배치 학습 데이터 크기
                     batch_size = min(batch_size, max_memory)