mlflow_storage

MLflow integration for model tracking and artifact storage.

Provides storage backend for managing MLflow experiments, runs, and artifacts. Handles serialization of trained models, logging of metrics and hyperparameters, and organization of local artifacts before uploading to MLflow tracking server.

Functions

normalize_tracking_uri(uri)

Normalize a tracking URI to a file:/// URI when it refers to a local path.

Classes

MLFlowStorage(**data)

MLflow storage backend for managing training runs and model artifacts.