SIENTIAPDE-1255: Implement MLFlow artifact management and model persistence

This commit introduces a new model_repository.py to handle MLFlow artifact generation and model persistence. It also updates the README to reflect this change and modifies training_repository.py to separate training and MLFlow operations.
This commit is contained in:
Bruno Domingues
2025-10-17 09:11:54 -03:00
parent a66b996cc1
commit 9e31ab679b
3 changed files with 18 additions and 916 deletions

View File

@@ -166,8 +166,9 @@ The Model Manager system uses a Temporal-based workflow architecture with clear
#### **Data Services (`model_manager/utils/`)**
- **Connectors Config**: Environment variable-based configuration management
- **Repository**: Data access layer for training operations
- **Repository**: Data access layer for training and MLFlow operations
- `training_repository.py`: Training business logic and operations
- `model_repository.py`: MLFlow artifact generation and model persistence
- **Models**: Data models and schemas
- `train_model_params.py`: Training parameters model
- `train_model_result.py`: Training result model
@@ -959,7 +960,8 @@ model_manager/
│ │ └── experiment_status.py # Experiment status enum
│ └── repository/ # Data access layer
│ ├── __init__.py
── training_repository.py # Training business logic
── training_repository.py # Training business logic
│ └── model_repository.py # MLFlow artifact management
├── metrics.py # Prometheus metrics definitions
└── __init__.py
```