MLflow Skill
Expert system for MLflow operations, tracking, and model management.
When to Use This Skill
This skill should be triggered when:
- Working with MLflow Tracking, Projects, Models, or Registry
- Looking up MLflow API references or feature documentation
- Implementing MLflow integration in machine learning pipelines
- Debugging MLflow-related errors or issues
- Seeking best practices for experiment tracking and model deployment
Quick Reference
Common Patterns
Pattern 1: Install MLflow
pip install mlflow
Pattern 2: Command Line Help
mlflow --help
Pattern 3: Start UI
mlflow ui
Pattern 4: referencing models
models:/<model_name>/<version>
models:/<model_name>/Production
Pattern 5: Log Model (PyTorch example)
mlflow.pytorch.log_model(model, "model")
Example Code Patterns
Example 1: Configure S3 Endpoint (Bash)
export MLFLOW_S3_ENDPOINT_URL=https://s3.us-west-1.amazonaws.com
Example 2: Set Bucket Owner (Bash)
export MLFLOW_S3_EXPECTED_BUCKET_OWNER=123456789012
Reference Files
This skill includes practical documentation in references/:
- ml.md - Focused guide for tracking, server setup, artifact store, and model registry.
Read specific sections of these files when the user needs detailed explanations.
usage
For Beginners
Start by reading the overview or getting started sections in references/ml.md to understand the core concepts of Runs, Experiments, and the Tracking Server.
For API Lookups
Search for the specific function or class name (e.g., mlflow.log_param, mlflow.sklearn.log_model) within references/ml.md to find parameters and usage examples.
For Troubleshooting
Check the reference documentation for configuration options and common pitfalls, especially regarding authentication and remote server connectivity.