ml-experiment

Track ML experiment results and store resolved Hydra configurations.

Updated Feb 6, 2026
One-click install
npx skills add https://github.com/nishide-dev/claude-code-ml-research --skill ml-experiment-nishide-dev
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: ml-experiment
Source: https://github.com/nishide-dev/claude-code-ml-research/tree/main/skills/ml-experiment
Command: npx skills add https://github.com/nishide-dev/claude-code-ml-research --skill ml-experiment-nishide-dev

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires matplotlib, pandas, and includes scripts (resource) components.

What problem does it solve?

This skill solves the problem of losing time and insight when managing many machine learning experiments by centralizing configs, metrics, and comparisons in a single workflow.

Core Features & Use Cases

  • Experiment config creation: Generate Hydra experiment configuration files (e.g., baseline, ablations, sweeps) to standardize what you run.
  • Result tracking and registry logging: Record experiment status, hyperparameters, and key metrics into an experiment_registry.json for later review.
  • Cross-experiment comparison & analysis: Compare multiple runs by metrics (accuracy, loss, runtime, and hyperparameters) and produce comparison plots for quick decision-making.
  • Reproducibility support: Save environment/package versions and git commit hashes, then rerun using the stored Hydra resolved config.

Quick Start

Run an experiment and have it logged to the registry by executing: python src/train.py experiment=vit_imagenet_finetuning

Frequently Asked Questions about ml-experiment

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

FAQPage Schema
How do I track ML experiment metrics and compare runs across different settings?

To track ML experiment metrics and compare runs, this skill logs hyperparameters and key metrics to an experiment registry, generating comparison plots to analyze accuracy, loss, and runtime across multiple runs for quick decision-making.

How do I make PyTorch Lightning training runs reproducible?

To make PyTorch Lightning training runs reproducible, this skill saves environment package versions, git commit hashes, and resolved Hydra configurations, allowing you to accurately rerun any stored experiment setup later.

Does this ML experiment tracking skill work with TensorBoard and MLflow?

Yes, the ML experiment tracking skill supports logging and metric-based selection using W&B, TensorBoard, or MLflow, integrating these platforms into your training workflow to manage experiment registry logging.

How do I generate Hydra configs for ML experiment sweeps and ablations?

You can generate Hydra configs for ML experiment sweeps and ablations by using this skill to standardize baseline configurations, creating structured files that standardize what you run before executing training.

What is the best way to compare ML hyperparameters across multiple training runs?

The best way to compare ML hyperparameters across multiple training runs is to record them into an experiment registry JSON file, using companion scripts to generate comparison plots and select models based on metrics.

Why do I need to log experiment status to a JSON registry file?

You need to log experiment status to a JSON registry file to centralize configs and metrics in a single workflow, preventing the loss of insight and time when managing many machine learning experiments.