fix histo Pt.9

Techtonique · Oct 28, 2024 · 4a530bb · 4a530bb
1 parent 3c32436
commit 4a530bb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mlsauce/booster/_booster_classifier.py b/mlsauce/booster/_booster_classifier.py
@@ -403,7 +403,7 @@ def fit(self, X, y, **kwargs):
         if isinstance(X, pd.DataFrame):
             X = X.values
 
-        if self.hist:
+        if self.hist == True:
             X, self.hist_bins_ = get_histo_features(X)
 
         if isinstance(y, pd.Series):
@@ -496,7 +496,7 @@ def predict_proba(self, X, **kwargs):
         if isinstance(X, pd.DataFrame):
             X = X.values
 
-        if self.hist:
+        if self.hist == True:
             X = get_histo_features(X, bins=self.hist_bins_)
 
         if self.degree is not None:

diff --git a/mlsauce/booster/_booster_regressor.py b/mlsauce/booster/_booster_regressor.py
@@ -273,7 +273,7 @@ def fit(self, X, y, **kwargs):
         if isinstance(X, pd.DataFrame):
             X = X.values
 
-        if self.hist:
+        if self.hist == True:
             X, self.hist_bins_ = get_histo_features(X)
 
         if isinstance(y, pd.Series):
@@ -360,7 +360,7 @@ def predict(self, X, level=95, method=None, histo=False, **kwargs):
         if isinstance(X, pd.DataFrame):
             X = X.values
 
-        if self.hist:
+        if self.hist == True:
             X = get_histo_features(X, bins=self.hist_bins_)
 
         if self.degree is not None: