Update dqn.py

komo135 · Mar 8, 2022 · 84c8242 · 84c8242
1 parent 7be3dfd
commit 84c8242
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/traderl/agent/dqn.py b/traderl/agent/dqn.py
@@ -100,7 +100,7 @@ def env(self):
         return x, y, atr
 
     def _build_model(self) -> nn.model.Model:
-        model = nn.build_model(self.model_name, self.x.shape[1:], 2, self.agent_name)
+        model = nn.build_model(self.model_name, self.x.shape[1:], 2, None, self.agent_name)
         model.compile(
             tf.keras.optimizers.Adam(self.lr, clipnorm=1.), loss=self.loss, steps_per_execution=100
         )
@@ -405,4 +405,4 @@ def train(self, epoch=40, batch_size=2056):
             self.target_model.set_weights(self.model.get_weights())
 
 
-__all__ = ["DQN"]
+__all__ = ["DQN"]