docs

rvandewater · Oct 17, 2024 · 619937c · 619937c
1 parent 5004ae4
commit 619937c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/icu_benchmarks/tuning/hyperparameters.py b/icu_benchmarks/tuning/hyperparameters.py
@@ -199,7 +199,8 @@ def choose_and_bind_hyperparameters_optuna(
     """Choose hyperparameters to tune and bind them to gin. Uses Optuna for hyperparameter optimization.
 
     Args:
-        sampler:
+        plot: Whether to plot hyperparameter importances.
+        sampler: The sampler to use for hyperparameter optimization.
         wandb: Whether we use wandb or not.
         load_cache: Load cached data if available.
         generate_cache: Generate cache data.