wip mlflow implementation

This commit is contained in:
2026-05-26 15:03:53 -04:00
parent 0e728cc193
commit b907a74525
10 changed files with 2190 additions and 26 deletions

View File

@@ -55,6 +55,9 @@ class SageMakerConfig(BaseModel):
class MlflowConfig(BaseModel):
mode: MlflowMode = MlflowMode.disabled
tracking_server_name: str | None = None
experiment_name: str = "qc-cli-training"
registered_model_name: str = "qc-cli-model"
register_trained_models: bool = True
artifact_prefix: str = "mlflow/"
tracking_server_size: MlflowServerSize = MlflowServerSize.small
mlflow_version: str | None = None