From 7970425431922641ff06b8d6d5bc360c6128ed15 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 5 Feb 2025 22:29:58 +0100 Subject: [PATCH] fix(qav3): reset df index after concat MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py | 3 ++- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index ff5dbcf..6ad5991 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py @@ -269,7 +269,8 @@ def objective( ) error = sklearn.metrics.root_mean_squared_error( - pd.concat([y_test_min, y_test_max]), pd.concat([y_pred_min, y_pred_max]) + pd.concat([y_test_min, y_test_max]).reset_index(drop=True), + pd.concat([y_pred_min, y_pred_max]).reset_index(drop=True), ) return error diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py index 33188d8..992696b 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py @@ -274,7 +274,8 @@ def objective( ) error = sklearn.metrics.root_mean_squared_error( - pd.concat([y_test_min, y_test_max]), pd.concat([y_pred_min, y_pred_max]) + pd.concat([y_test_min, y_test_max]).reset_index(drop=True), + pd.concat([y_pred_min, y_pred_max]).reset_index(drop=True), ) return error -- 2.43.0