mcpField
C
classification-helper

classification-helper

github.com/dkyazzentwatwa数据与数据库12454 次安装29 周前更新
代码仓库

关于 classification-helper

classification-helper 是用于训练和评估分类模型的命令行技能，可比较分类器、调整超参数并导出训练好的模型。

主要功能

  1. 01比较分类模型
  2. 02调整超参数
  3. 03使用多种指标评估
  4. 04执行 K 折验证
  5. 05检查混淆矩阵和特征重要性
  6. 06导出训练好的模型

适用场景

  1. 01比较训练数据集上的分类器
  2. 02通过网格或随机搜索调参
  3. 03检查分类错误和预测特征

安装

复制下面的命令，在终端中运行。

$npx skills add https://github.com/dkyazzentwatwa/chatgpt-skills --skill classification-helper

安装后在 Claude Code 中输入/classification-helper确认已加载。

附带文件scripts/classification_helper.pyscripts/requirements.txt

详细介绍

分类辅助工具

使用自动模型选择来训练和评估分类模型。

功能

  • 自动模型选择：比较多个分类器
  • 超参数调优：网格搜索／随机搜索
  • 评估指标：准确率、精确率、召回率、F1、ROC-AUC
  • 交叉验证：K 折验证
  • 混淆矩阵：详细的错误分析
  • 特征重要性：最具预测性的特征
  • 模型导出：保存训练好的模型

CLI 用法

python classification_helper.py --data train.csv --target class --test test.csv --output model.pkl

依赖项

  • scikit-learn>=1.3.0
  • pandas>=2.0.0
  • numpy>=1.24.0
  • matplotlib>=3.7.0
  • seaborn>=0.12.0

评论

支持 Markdown