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.
|
||||
prediction_data = wrapper.retrain(data)
|
||||
prediction_data.rename(columns={target: 'prediction'}, inplace=True)
|
||||
|
||||
# Merge prediction data with retrain data
|
||||
evaluation_data = pd.merge(
|
||||
|
||||
Reference in New Issue
Block a user