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:
@@ -529,6 +529,7 @@ class MLFlow(SientiaMonitoring):
|
|||||||
|
|
||||||
# Keep heavy model fitting outside MLflow run timing.
|
# Keep heavy model fitting outside MLflow run timing.
|
||||||
prediction_data = wrapper.retrain(data)
|
prediction_data = wrapper.retrain(data)
|
||||||
|
prediction_data.rename(columns={target: 'prediction'}, inplace=True)
|
||||||
|
|
||||||
# Merge prediction data with retrain data
|
# Merge prediction data with retrain data
|
||||||
evaluation_data = pd.merge(
|
evaluation_data = pd.merge(
|
||||||
|
|||||||
Reference in New Issue
Block a user