Release

Author: cerlymarco

lineartree/lineartree.py

@@ -1544,20 +1544,8 @@ def predict_proba(self, X):
             The class probabilities of the input samples. The order of the
             classes corresponds to that in the attribute :term:`classes_`.
         """
-        check_is_fitted(self, attributes='base_estimator_')
-
-        X = self._validate_data(
-            X,
-            reset=False,
-            accept_sparse=True,
-            dtype='float32',
-            force_all_finite=True,
-            ensure_2d=True,
-            allow_nd=False,
-            ensure_min_features=self.n_features_in_
-        )
 
-        pred = self._sigmoid(self.base_estimator_.predict(X))
+        pred = self._sigmoid(self.decision_function(X))
         proba = np.zeros((X.shape[0], 2))
         proba[:, 0] = 1 - pred
         proba[:, 1] = pred

setup.py

@@ -3,7 +3,7 @@
 
 HERE = pathlib.Path(__file__).parent
 
-VERSION = '0.3.2'
+VERSION = '0.3.3'
 PACKAGE_NAME = 'linear-tree'
 AUTHOR = 'Marco Cerliani'
 AUTHOR_EMAIL = '[email protected]'