Update cyclic_boosting.py

sktime · May 16, 2024 · 39d0e49 · 39d0e49
1 parent dad8793
commit 39d0e49
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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,