Release

Author: cerlymarco

lineartree/_classes.py

@@ -335,7 +335,7 @@ def _grow(self, X, y, weights=None):
 
         # extract quantiles
         bins = np.linspace(0, 1, self.max_bins)[1:-1]
-        bins = np.quantile(X, bins, axis=0, interpolation='midpoint')
+        bins = np.quantile(X, bins, axis=0)
         bins = list(bins.T)
         bins = [np.unique(X[:, c]) if c in self._categorical_features
                 else np.unique(q) for c, q in enumerate(bins)]

setup.py

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