From de1e11a143f800b206dcccaef71109fb1b1f938b Mon Sep 17 00:00:00 2001 From: busyyang Date: Sun, 2 Feb 2020 17:05:31 +0800 Subject: [PATCH] Update Bases.py --- Util/Bases.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: