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

feat: enhance training workflow with model metadata loading and refactor data handling

Browse Source 
- Introduced a new activity to load model metadata from the model store.
- Refactored training logic to utilize new model metadata and improved parameter handling.
- Updated the `TrainModelParams` class to include additional fields for model configuration.
- Replaced deprecated utility functions with a custom train-test split implementation.
- Removed unused utility functions and cleaned up the data manager repository.
- Adjusted experiment tracking to include model-specific metadata in notifications.
This commit is contained in:
vitor-aignosi
2026-03-24 14:39:28 -03:00
parent cf5111e520
commit 342a02d6f7
13 changed files with 242 additions and 465 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/run_training_test.py
View File

@@ -341,7 +341,6 @@ def run_local_pipeline(
metadata={'source': 'run_local_pipeline', 'scenario': scenario_name},
)
train_result = data_manager_repository.compute_regression_metrics(
train_params,
train_result,
)
except Exception as e: