From 251ec1ad2566374f2770a1279c85d77d66a6d00a Mon Sep 17 00:00:00 2001
From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= <jerome.benoit@piment-noir.org>
Date: Mon, 27 Jan 2025 13:39:11 +0100
Subject: [PATCH] refactor: ensure metric rmse is used as LightGBM optuna
 metric
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---
 .../user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
index b7540a4..5fe43c6 100644
--- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
+++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
@@ -243,6 +243,7 @@ def objective(trial, X, y, weights, X_test, y_test, params):
         y,
         sample_weight=weights,
         eval_set=[(X_test, y_test)],
+        eval_metric="rmse",
         callbacks=[optuna.integration.LightGBMPruningCallback(trial, "rmse")],
     )
     y_pred = model.predict(X_test)
-- 
2.43.0