custom info to tensorboard_metrics

This commit is contained in:
initrv
2022-12-11 15:37:45 +03:00
parent 58604c747e
commit cb8fc3c8c7
5 changed files with 10 additions and 15 deletions

View File

@@ -46,9 +46,9 @@ class Base4ActionRLEnv(BaseEnvironment):
self._done = True
self._update_unrealized_total_profit()
step_reward = self.calculate_reward(action)
self.total_reward += step_reward
self.tensorboard_metrics[self.actions._member_names_[action]] += 1
trade_type = None
if self.is_tradesignal(action):