From da9db322dac1b77766f9a8e90600b905cda64b64 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Mon, 20 Mar 2023 18:58:55 +0100 Subject: [PATCH] Fix sklearn requirement --- src/fimdlp/mdlp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/fimdlp/mdlp.py b/src/fimdlp/mdlp.py index 3809b4b..5a8ea8c 100644 --- a/src/fimdlp/mdlp.py +++ b/src/fimdlp/mdlp.py @@ -98,7 +98,7 @@ class FImdlp(TransformerMixin, BaseEstimator): self._update_params(X, y) self.X_ = X self.y_ = y - self.efective_min_length = ( + self.efective_min_length_ = ( self.min_length if self.min_length > 1 else int(self.min_length * X.shape[0]) @@ -114,7 +114,7 @@ class FImdlp(TransformerMixin, BaseEstimator): def _fit_discretizer(self, feature): if feature in self.features_: self.discretizer_[feature] = CFImdlp( - min_length=self.efective_min_length, + min_length=self.efective_min_length_, max_depth=self.max_depth, max_cuts=self.max_cuts, )