Dataset Splitter 技能:按类别划分训练、验证和测试集
EnglishClaude Code 安装步骤
- 从 GitHub 固定提交获取完整的 `plugins/ai-ml/dataset-splitter/skills/splitting-datasets` 目录。
- 复制到目标项目的 `.claude/skills/`,确认 `.claude/skills/splitting-datasets/SKILL.md`,再从项目根目录启动 Claude Code,输入 `/splitting-datasets`。
中文摘要
用 Claude Code 将有标签的 CSV 分成训练、验证和测试数据,固定随机种子、维持各集合的类别比例并核对 ID 交集。
SKILL.md 中文导读
可复现示例
创建 1,000 条合成记录,800 条标签 A、200 条标签 B。先用 train_test_split 按 label 分层划出 70%,再将其余 300 条按 label 等分,写出 train.csv、val.csv、test.csv。工作步骤
- 先以固定随机种子按类别分层划出训练集,再对余下部分做第二次分层,形成验证集与测试集。
- 核对集合行数、各标签占比、ID 交集和总行数。将同一 ID 组或时间相关的数据用适合的数据划分策略处理。
验收方法
三个文件分别为 700、150、150 行,标签 B 各占 20%,且 row_id 两两无交集。
GitHub 原文 · 更新日期
适用环境
准备一份每行有唯一 row_id 与 label 的合成 CSV。示例使用 pandas 和 scikit-learn。
常见问题
同一客户有多行记录,按标签随机分层够用吗?
先按客户 ID 分组隔离,避免同一客户横跨多个集合。完成后另外报告类别比例。
相关指南
相关推荐
groq-install-auth 技能
jeremylongshore/claude-code-plugins-plus-skills
anima-common-errors 技能
jeremylongshore/claude-code-plugins-plus-skills
apple-notes-hello-world 技能
jeremylongshore/claude-code-plugins-plus-skills
castai-hello-world 技能
jeremylongshore/claude-code-plugins-plus-skills
cohere-local-dev-loop 技能
jeremylongshore/claude-code-plugins-plus-skills
figma-security-basics 技能
jeremylongshore/claude-code-plugins-plus-skills