Experiment Tracking Setup

Configure ML experiment tracking with MLflow or Weights & Biases, including environment setup and code for logging parameters, metrics, and artifacts.

Overview

This skill streamlines the process of setting up experiment tracking for machine learning projects. It automates environment configuration, tool initialization, and provides code examples to get you started quickly.

How It Works

1. Analyze Context: The skill analyzes the current project context to determine the appropriate experiment tracking tool (MLflow or W&B) based on user preference or existing project configuration.
2. Configure Environment: It configures the environment by installing necessary Python packages and setting environment variables.
3. Initialize Tracking: The skill initializes the chosen tracking tool, potentially starting a local MLflow server or connecting to a W&B project.
4. Provide Code Snippets: It provides code snippets demonstrating how to log experiment parameters, metrics, and artifacts within your ML code.

When to Use This Skill