add tensorboard category

This commit is contained in:
initrv
2023-03-12 01:32:55 +03:00
parent b23841fbfe
commit 82cb107520
7 changed files with 26 additions and 21 deletions

View File

@@ -49,7 +49,7 @@ class Base5ActionRLEnv(BaseEnvironment):
self._update_unrealized_total_profit()
step_reward = self.calculate_reward(action)
self.total_reward += step_reward
self.tensorboard_log(self.actions._member_names_[action])
self.tensorboard_log(self.actions._member_names_[action], category="actions")
trade_type = None
if self.is_tradesignal(action):