From 4fb775981b2ab8fa57ccd3ef22d3f4f2e9149e25 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Mon, 17 Jan 2022 08:02:33 +0100
Subject: XGBoost: Input data must be two-dimensional

---
 lib/parameters.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/parameters.py b/lib/parameters.py
index c586d32..4b6dc2c 100644
--- a/lib/parameters.py
+++ b/lib/parameters.py
@@ -945,7 +945,7 @@ class ModelAttribute:
             fit_parameters, category_to_index, ignore_index = param_to_ndarray(
                 parameters, with_nan=False, categorial_to_scalar=categorial_to_scalar
             )
-            xgb.fit(fit_parameters, data)
+            xgb.fit(fit_parameters, np.reshape(data, (-1, 1)))
             self.model_function = df.SKLearnRegressionFunction(
                 np.mean(data), xgb, category_to_index, ignore_index
             )
-- 
cgit v1.2.3