diff --git a/Util/Bases.py b/Util/Bases.py index b1d3f29..ef5edce 100644 --- a/Util/Bases.py +++ b/Util/Bases.py @@ -254,7 +254,7 @@ def f1_score(y, y_pred): @staticmethod def _multi_clf(x, clfs, task, kwargs, stack=np.vstack, target="single"): if target != "parallel": - return np.array([clf.predict(x) for clf in clfs], dtype=np.float32).T + return np.array([clf.predict(x) for clf in clfs]).T n_cores = kwargs.get("n_cores", 2) n_cores = multiprocessing.cpu_count() if n_cores <= 0 else n_cores if n_cores == 1: