From 39d0e49ced2966214ca74c2cfc6a9f38bf54b113 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Franz=20Kir=C3=A1ly?= Date: Thu, 16 May 2024 16:29:06 +0100 Subject: [PATCH] Update cyclic_boosting.py --- skpro/regression/cyclic_boosting.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/skpro/regression/cyclic_boosting.py b/skpro/regression/cyclic_boosting.py index 19620db78..230f0c5f7 100644 --- a/skpro/regression/cyclic_boosting.py +++ b/skpro/regression/cyclic_boosting.py @@ -284,9 +284,9 @@ def _predict_proba(self, X): # Johnson Quantile-Parameterized Distributions params = { "alpha": self.alpha, - "qv_low": self.quantile_values[0], - "qv_median": self.quantile_values[1], - "qv_high": self.quantile_values[2], + "qv_low": self.quantile_values[0].reshape(-1, 1), + "qv_median": self.quantile_values[1].reshape(-1, 1), + "qv_high": self.quantile_values[2].reshape(-1, 1), "lower": self.lower, "upper": self.upper, "version": self.version,