Create a dataset using an agent (#306)

Generate a dataset using an agent (allowing to select between this and a random dataset)
2025-12-17 19:20:19 +01:00 · 2019-05-28 09:34:49 +03:00
parent 342b7184bc
commit 9e9c4fd332
26 changed files with 351 additions and 111 deletions
--- a/rl_coach/agents/dqn_agent.py
+++ b/rl_coach/agents/dqn_agent.py
@@ -49,6 +49,7 @@ class DQNNetworkParameters(NetworkParameters):
        self.batch_size = 32
        self.replace_mse_with_huber_loss = True
        self.create_target_network = True
+        self.should_get_softmax_probabilities = False


 class DQNAgentParameters(AgentParameters):