hyperparameters tuning

agi-brain · Jan 16, 2025 · 978021e · 978021e
1 parent 798b075
commit 978021e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/examples/hyperparameter_tuning/tune_dqn.py b/examples/hyperparameter_tuning/tune_dqn.py
@@ -6,7 +6,7 @@
                             running_steps=1000,
                             test_episodes=2)
 
-selected_hyperparameters = tuner.select_hyperparameter(['learning_rate'])
+selected_hyperparameters = tuner.select_hyperparameter(['learning_rate', 'gamma'])
 
 overrides = {
     'learning_rate': [0.001, 0.0001, 0.00001],  # Categorical: Search in the set of {0.001, 0.0001, 0.00001}.

diff --git a/setup.py b/setup.py
@@ -49,6 +49,7 @@
         "tune": [
             "optuna>=4.1.0",
             "optuna-dashboard>=0.17.0",
+            "plotly>=5.24.1",
         ],
         "atari": ["gymnasium[accept-rom-license]==0.28.1",
                   "gymnasium[atari]==0.28.1",