added structuredPICNN corrected initialization bug

supsi-dacd-isaac · Jan 9, 2024 · 896b686 · 896b686
1 parent a37bd16
commit 896b686
Show file tree

Hide file tree

Showing 3 changed files with 294 additions and 64 deletions.
diff --git a/pyforecaster/dictionaries.py b/pyforecaster/dictionaries.py
@@ -31,5 +31,6 @@ def picnn_param_space(trial):
 HYPERPAR_MAP = {'LinearForecaster': linear_param_space,
                 'LGBForecaster': lgb_param_space,
                 'LGBMHybrid': lgb_param_space,
-                "PICNN": picnn_param_space}
+                "PICNN": picnn_param_space,
+                "StructuredPICNN": picnn_param_space}