wip mlflow implementation
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user