diff --git a/freqtrade/freqai/RL/Base4ActionRLEnv.py b/freqtrade/freqai/RL/Base4ActionRLEnv.py index 478507639..ce94ff542 100644 --- a/freqtrade/freqai/RL/Base4ActionRLEnv.py +++ b/freqtrade/freqai/RL/Base4ActionRLEnv.py @@ -292,7 +292,8 @@ class Base4ActionRLEnv(gym.Env): pnl = self.get_unrealized_profit() if self._position in (Positions.Long, Positions.Short): - self._total_profit *= (1 + pnl) + #self._total_profit *= (1 + pnl) + self._total_profit += pnl self._profits.append((self._current_tick, self._total_profit)) self.close_trade_profit.append(pnl) diff --git a/freqtrade/freqai/RL/Base5ActionRLEnv.py b/freqtrade/freqai/RL/Base5ActionRLEnv.py index b93d6e6ff..0441a10b9 100644 --- a/freqtrade/freqai/RL/Base5ActionRLEnv.py +++ b/freqtrade/freqai/RL/Base5ActionRLEnv.py @@ -295,7 +295,8 @@ class Base5ActionRLEnv(gym.Env): pnl = self.get_unrealized_profit() if self._position in (Positions.Long, Positions.Short): - self._total_profit *= (1 + pnl) + #self._total_profit *= (1 + pnl) + self._total_profit += pnl self._profits.append((self._current_tick, self._total_profit)) self.close_trade_profit.append(pnl)