From 6ef82dd8b6b92c0e0eb02e47d115f3272082fec6 Mon Sep 17 00:00:00 2001 From: Mark Regan Date: Thu, 27 Oct 2022 12:41:12 +0100 Subject: [PATCH] minor change to return --- .../freqai/base_models/FreqaiMultiOutputClassifier.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py b/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py index a4a8ddfcb..ce4b6ec84 100644 --- a/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py +++ b/freqtrade/freqai/base_models/FreqaiMultiOutputClassifier.py @@ -97,8 +97,10 @@ class FreqaiMultiOutputClassifier(MultiOutputClassifier): ``n_classes``) for that particular output. """ check_is_fitted(self) - results = np.hstack([estimator.predict_proba(X) for estimator in self.estimators_]) - return np.squeeze(results) + results = np.squeeze(np.hstack( + [estimator.predict_proba(X) for estimator in self.estimators_] + )) + return results def predict(self, X): """Predict multi-output variable using model for each target variable. @@ -122,6 +124,6 @@ class FreqaiMultiOutputClassifier(MultiOutputClassifier): delayed(e.predict)(X) for e in self.estimators_ ) - results = np.asarray(y).T + results = np.squeeze(np.asarray(y).T) - return np.squeeze(results) + return results