mlflow

MLflow integration for model tracking and storage.

Provides integration with MLflow for model lifecycle management, experiment tracking, and model registry functionality. This package enables OpenSTEF models to be stored, versioned, and tracked using MLflow’s model registry.

Note

This package requires MLflow to be installed as an optional dependency. MLflow integration is particularly useful for production deployments requiring model versioning, experiment tracking, and centralized storage.

Submodules

openstef_models.integrations.mlflow.mlflow_storage	MLflow integration for model tracking and artifact storage.
openstef_models.integrations.mlflow.mlflow_storage_callback	MLflow integration for tracking and storing forecasting workflows.

Classes

MLFlowStorage(**data)	MLflow storage backend for managing training runs and model artifacts.
MLFlowStorageCallback(**data)	MLFlow callback for logging forecasting workflow events.