Add missing params to ShapeletImportance estimator

wildboar-foundation · Nov 12, 2024 · 99d7589 · 99d7589
1 parent e3f2b11
commit 99d7589
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/wildboar/explain/_importance.py b/src/wildboar/explain/_importance.py
@@ -1015,6 +1015,8 @@ def fit(self, estimator, X, y, sample_weight=None):
             metric=self.metric,
             min_shapelet_size=self.min_shapelet_size,
             max_shapelet_size=self.max_shapelet_size,
+            coverage_probability=self.coverage_probability,
+            variability=self.variability,
             random_state=random_state.randint(np.iinfo(np.int32).max),
         ).fit(X)