mlflow-experiment-tracker

Solid

MLflow integration skill for experiment tracking, model registry, and artifact management. Enables LLMs to log experiments, compare runs, manage model lifecycle, and retrieve artifacts through the MLflow API.

AI & Automation 814 stars 53 forks Updated today MIT

Install

View on GitHub

Quality Score: 95/100

Stars 20%
97
Recency 20%
100
Frontmatter 20%
70
Documentation 15%
100
Issue Health 10%
50
License 10%
100
Description 5%
100

Skill Content

# MLflow Experiment Tracker Integrate with MLflow for comprehensive ML experiment tracking, model registry operations, and artifact management. ## Overview This skill provides capabilities for interacting with MLflow's tracking server and model registry. It enables automated experiment logging, run comparison, model versioning, and artifact retrieval within ML workflows. ## Capabilities ### Experiment Management - Create and manage experiments - Start and end runs programmatically - Set experiment tags and descriptions - List and search experiments ### Parameter and Metric Logging - Log hyperparameters for reproducibility - Track metrics during training (loss, accuracy, etc.) - Log batch metrics with timestamps - Set run tags for organization ### Artifact Management - Log model artifacts (serialized models, checkpoints) - Store datasets and data samples - Save plots and visualizations - Retrieve artifacts from completed runs ### Model Registry Operations - Register trained models - Manage model versions - Transition models between stages (Staging, Production, Archived) - Add model descriptions and tags ### Run Comparison and Analysis - Compare metrics across runs - Search runs by parameters/metrics - Retrieve best performing runs - Generate comparison visualizations ## Prerequisites ### MLflow Installation ```bash pip install mlflow>=2.0.0 ``` ### MLflow Tracking Server Configure tracking URI: ```python import mlflow mlflow.set_tracking_uri("http://localhost:5000"...

Details

Author
a5c-ai
Repository
a5c-ai/babysitter
Created
4 months ago
Last Updated
today
Language
JavaScript
License
MIT

Related Skills