-

LocalCascadeEnsemble · Sep 11, 2022 · b746871 · b746871
1 parent cafa180
commit b746871
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lce/_lcetree.py b/lce/_lcetree.py
@@ -1210,7 +1210,7 @@ def _predict(node, X, y_pred_final=None):
         X = np.concatenate((index, X), axis=1)
         y_pred = _predict(self.tree, X, None)
         y_pred = y_pred[y_pred[:, 0].argsort()]
-        y_pred = np.squeeze(y_pred[:, 1:])
+        y_pred = y_pred[:, 1:]
         return y_pred
 
     def set_params(self, **params):

diff --git a/lce/_version.py b/lce/_version.py
@@ -1 +1 @@
-__version__ = "0.2.9"
+__version__ = "0.3.0"