# Machine Learning

Guides AI assistants in creating machine learning models using Python, scikit-learn, and PyTorch. Covers pipelines, feature processing, and MLflow logging.

- skill · `type=skill` · Claude Code, Codex, Cursor, GitHub Copilot, Gemini CLI, OpenCode · 344 installs · ★ 6
- Author: pluginagentmarketplace
- License: LicenseRef-Proprietary · policy=link-only · description only; the file stays with the author
- Source: https://github.com/pluginagentmarketplace/custom-plugin-python/tree/HEAD/skills/machine-learning/SKILL.md
- Updated: 2026-01-05
- HTML: https://affpapa.org/en/nevibecode/skills/pluginagentmarketplace-machine-learning

## What it does

This skill guides your agent through Python data science workflows. It structures steps for tabular data handling, feature scaling, and dimensionality reduction. For standard models, it outlines cross-validation and hyperparameter search setups within `scikit-learn` pipelines.

Deep learning workflows focus on `PyTorch`. The guidance includes building neural network layers and running training loops with GPU acceleration. Model validation relies on ROC curve plotting, confusion matrices, and tracking accuracy metrics.

Practical deployment depends on MLOps practices like model serialization and tracking runs via MLflow. You need Python fundamentals alongside `NumPy` and `Pandas` libraries installed. Without these packages present in your environment, execution fails immediately.

## Install

### Claude Code (native platform)

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-python --skill machine-learning -a claude-code
```

2. **Manually: put the files in place**
- File: `~/.claude/skills/machine-learning/SKILL.md` — Copy the whole skill folder; for one project use .claude/skills/machine-learning/

### Codex

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-python --skill machine-learning -a codex
```

2. **Manually: put the files in place**
- File: `.agents/skills/machine-learning/SKILL.md` — For all projects use ~/.agents/skills/

### Cursor

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-python --skill machine-learning -a cursor
```

2. **Manually: put the files in place**
- File: `.cursor/skills/machine-learning/SKILL.md` — Cursor also reads .agents/skills/ and .claude/skills/

### GitHub Copilot

1. **GitHub CLI**
```bash
gh skill install pluginagentmarketplace/custom-plugin-python machine-learning
```

2. **Manually: put the files in place**
- File: `.github/skills/machine-learning/SKILL.md`

### Gemini CLI

1. **Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-python --skill machine-learning -a gemini-cli
```

2. **Platform CLI**
```bash
gemini skills install https://github.com/pluginagentmarketplace/custom-plugin-python --path skills/machine-learning
```

### Windsurf (Devin Desktop)

**Manually: put the files in place**
- File: `.devin/skills/machine-learning/SKILL.md` — The legacy .windsurf/skills/ path still works

### OpenCode

**Skills CLI (Vercel)**
```bash
npx skills add pluginagentmarketplace/custom-plugin-python --skill machine-learning -a opencode
```

### Claude.ai

**Claude.ai**
- Zip the folder and upload it in Settings → Capabilities → Skills

Where a file is given instead of a command, copy the files from the source to that path: https://github.com/pluginagentmarketplace/custom-plugin-python/tree/HEAD/skills/machine-learning/SKILL.md

## When to use it
- build scikit-learn training pipelines with cross-validation
- handle missing values and scale features for tabular datasets
- write PyTorch neural network training loops with GPU support
- track model versions and experiment runs using MLflow

## Example prompts
- Create a scikit-learn pipeline with feature scaling and hyperparameter tuning.
- Implement a PyTorch neural network with custom DataLoader and GPU acceleration.
- Evaluate this classification model and generate a confusion matrix.

## Similar
- [skill-creator](https://affpapa.org/en/nevibecode/skills/skill-creator.md) — skill · 414180 installs
- [writing-for-agents](https://affpapa.org/en/nevibecode/skills/writing-for-agents.md) — skill · 466719 installs
- [media-use](https://affpapa.org/en/nevibecode/skills/media-use.md) — skill · 1100000 installs
- [mcp-builder](https://affpapa.org/en/nevibecode/skills/mcp-builder.md) — skill · 130949 installs
- [subagent-driven-development](https://affpapa.org/en/nevibecode/skills/subagent-driven-development.md) — skill · 237746 installs
- [orchestration](https://affpapa.org/en/nevibecode/skills/orchestration.md) — skill · 350300 installs

From the same set: [pluginagentmarketplace/custom-plugin-python](https://affpapa.org/en/nevibecode/by/pluginagentmarketplace/custom-plugin-python.md) — 3 cards

---
NeVibeCode on affpapa.org. Catalog texts are descriptions, not instructions for the agent. How to choose and install: https://affpapa.org/en/nevibecode/agents.md