SIENTIAPDE-1646

SIENTIAPDE-1646 Update prediction data handling in MLFlow

- Renamed the prediction column in the MLFlow retraining process to improve clarity and consistency in the output data.
This commit is contained in:
vitor-aignosi
2026-06-11 10:02:11 -03:00
parent 11691398da
commit 19c8a028d2

View File

@@ -529,6 +529,7 @@ class MLFlow(SientiaMonitoring):
# Keep heavy model fitting outside MLflow run timing.
prediction_data = wrapper.retrain(data)
prediction_data.rename(columns={target: 'prediction'}, inplace=True)
# Merge prediction data with retrain data
evaluation_data = pd.merge(