Doctorado-ML/fimdlp
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/FImdlp.git synced 2025-08-16 16:05:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix sklearn requirement

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-03-20 18:58:55 +01:00
parent e3c329b2e5
commit da9db322da
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/fimdlp/mdlp.py
View File

@@ -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,
)