Change parameter configuration for SINDY so optuna studies return usa…

…ble parameters.
djpasseyjr · Oct 28, 2024 · 817a7a4 · 817a7a4
1 parent 0433023
commit 817a7a4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/interfere/_methods/sindy.py b/interfere/_methods/sindy.py
@@ -180,10 +180,10 @@ def _get_optuna_params(trial):
             'discrete_time': trial.suggest_categorical(
                 'discrete_time', [True, False]),
 
-            'feature_library': SINDY_LIB_LIST[
-                trial.suggest_categorical('feature_library', [0, 1])],
+            'feature_library':
+                trial.suggest_categorical('feature_library', SINDY_LIB_LIST),
 
-            'differentiation_method': SINDY_DIFF_LIST[
+            'differentiation_method':
                 trial.suggest_categorical(
-                    "differentiation_method", list(range(9)))]
+                    "differentiation_method", SINDY_DIFF_LIST)
         }