gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add run_name to TrainModelResult for enhanced tracking

Browse Source 
- 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
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
model_manager/utils/repository/data_manager_repository.py
View File

@@ -200,7 +200,9 @@ class DataManagerRepository(SientiaMonitoring):
metadata, 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: def _as_series(self, pred: pd.DataFrame | pd.Series) -> pd.Series:
if isinstance(pred, pd.Series): if isinstance(pred, pd.Series):