From bab0fdf4adfc4ed96c296e21aa5056371ccba891 Mon Sep 17 00:00:00 2001 From: DJ Passey Date: Wed, 15 Jan 2025 18:33:31 -0500 Subject: [PATCH] Changed sindy optuna hyper parameter function to use types for feature library. --- interfere/_methods/sindy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/interfere/_methods/sindy.py b/interfere/_methods/sindy.py index 891a61c..65ddfc4 100644 --- a/interfere/_methods/sindy.py +++ b/interfere/_methods/sindy.py @@ -12,7 +12,7 @@ # Lists of hyper parameters for optimization. -SINDY_LIB_LIST = [ps.PolynomialLibrary(), ps.FourierLibrary()] +SINDY_LIB_LIST = [ps.PolynomialLibrary, ps.FourierLibrary] SINDY_DIFF_LIST = [ ps.SINDyDerivative(kind='finite_difference', k=1), @@ -44,7 +44,7 @@ def __init__(self, # Optionally accept types for feature library. if isinstance(feature_library, type): feature_library = feature_library() - + # Differentiation method and feature library must be copied so that # their internal state doesn't carry over across different fits. if differentiation_method is not None: