querying-mlflow-metrics

Fetch aggregated MLflow metrics with configurable server URL, experiment IDs, and aggregations.

1|1|Updated Oct 1, 2025
One-click install
npx skills add https://github.com/mkgs-databricks-demos/synthea-on-fhir --skill querying-mlflow-metrics-mkgs-databricks-demos
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: querying-mlflow-metrics
Source: https://github.com/mkgs-databricks-demos/synthea-on-fhir/tree/main/.cursor/skills/querying-mlflow-metrics
Command: npx skills add https://github.com/mkgs-databricks-demos/synthea-on-fhir --skill querying-mlflow-metrics-mkgs-databricks-demos

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes scripts (resource) components.

What problem does it solve?

Fetch aggregated metrics from MLflow tracking servers, including token usage, latency, trace counts, and quality evaluations, to empower data-driven monitoring and cost awareness.

Core Features & Use Cases

  • Retrieve aggregated metrics (total_tokens, latency, trace_count, input_tokens, output_tokens) from MLflow trackers.
  • Analyze token usage, latency, and quality assessments across experiments to optimize prompts and costs.
  • Query trace statistics and usage trends to inform model performance and experiment planning.
  • Use case: Monitor model deployments across multiple experiments and compare resource usage over time.

Quick Start

Run python scripts/fetch_metrics.py with the MLflow server URL and experiment IDs to retrieve the requested metrics.

Frequently Asked Questions about querying-mlflow-metrics

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I query aggregated MLflow metrics like token usage and latency?

You can fetch aggregated MLflow metrics such as token usage and latency by running a script with the MLflow server URL and experiment IDs. It retrieves requested metrics with optional dimensions and time-bucketing for on-demand analysis.

What MLflow metrics can I retrieve for model monitoring and cost analysis?

You can retrieve metrics including total_tokens, input_tokens, output_tokens, latency, trace_count, and quality evaluations. These aggregated metrics support monitoring model activity, diagnosing performance, and auditing costs across experiments.

How do I monitor token usage trends across multiple MLflow experiments?

To monitor token usage trends across multiple experiments, specify the MLflow server URL and target experiment IDs. The script aggregates token counts and trace statistics to inform model performance and experiment planning.

Does this approach require any external dependencies to fetch MLflow trace statistics?

No external dependencies are required. The Skill operates independently using bundled scripts to fetch trace statistics and usage trends directly from your MLflow tracking servers on-demand.

Can I apply time-bucketing and custom aggregations when querying MLflow metrics?

Yes, you can specify desired aggregations, optional dimensions, and apply time-bucketing. This allows you to analyze usage trends and quality assessments over specified intervals across your MLflow experiments.