From 817a7a4dd0b70665f0b7ca4428bf3dfa4a5ea882 Mon Sep 17 00:00:00 2001 From: DJ Passey Date: Mon, 28 Oct 2024 16:36:29 -0400 Subject: [PATCH] Change parameter configuration for SINDY so optuna studies return usable parameters. --- interfere/_methods/sindy.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/interfere/_methods/sindy.py b/interfere/_methods/sindy.py index 3ad72d4..6424d26 100644 --- 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) }