feat: add run_name to TrainModelResult for enhanced tracking

- Introduced run_name parameter in TrainModelResult to improve identification of training runs based on model type, name, and experiment run ID.
This commit is contained in:
vitor-aignosi
2026-04-16 09:34:36 -03:00
parent e658c3036f
commit c75a5921f3

View File

@@ -200,7 +200,9 @@ class DataManagerRepository(SientiaMonitoring):
metadata,
)
return TrainModelResult(params=params, train_data=train_data, val_data=val_data)
run_name = f'train_model_{params.model_type}_{params.model_name}_{params.experiment_run_id}'
return TrainModelResult(params=params, train_data=train_data, val_data=val_data, run_name=run_name)
def _as_series(self, pred: pd.DataFrame | pd.Series) -> pd.Series:
if isinstance(pred, pd.Series):