fix(qav3): reset df index after concat

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
index ff5dbcf4c3146080a375b95cccdb7f7138c018fd..6ad5991b17d749ee3b29e8880af8d5e5c8b85e01 100644 (file)
--- 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 33188d81e5dc1a35b5e4adbd4362739472d9e1d5..992696bf552bba1833563887491e6bc22733a1d1 100644 (file)
--- 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