fix self.degree at inference

Techtonique · May 8, 2024 · 663463c · 663463c
1 parent 2c0c29c
commit 663463c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/mlsauce/booster/_booster_classifier.py b/mlsauce/booster/_booster_classifier.py
@@ -290,7 +290,7 @@ def predict_proba(self, X, **kwargs):
         if isinstance(X, pd.DataFrame):
             X = X.values
 
-        if self.degree > 0:
+        if self.degree is not None:
             X = self.poly_.transform(X)
 
         if self.n_clusters > 0:

diff --git a/mlsauce/booster/_booster_regressor.py b/mlsauce/booster/_booster_regressor.py
@@ -298,7 +298,7 @@ def predict(self, X, level=95, method=None, **kwargs):
         if isinstance(X, pd.DataFrame):
             X = X.values
 
-        if self.degree > 0:
+        if self.degree is not None:
             X = self.poly_.transform(X)
 
         if self.n_clusters > 0:

diff --git a/setup.py b/setup.py
@@ -37,7 +37,7 @@
 MAINTAINER_EMAIL = '[email protected]'
 LICENSE = 'BSD3 Clause Clear'
 
-__version__ = '0.18.3'
+__version__ = '0.18.4'
 
 VERSION = __version__