mcpField
C
classification-helper

classification-helper

github.com/dkyazzentwatwaデータ・データベース12454 回インストール29 週間前に更新
リポジトリ

classification-helper について

classification-helper は、分類モデルの学習と評価に使うコマンドラインスキルです。分類器の比較やハイパーパラメーター調整を行い、学習済みモデルを出力できます。

主な機能

  1. 01分類モデルの比較
  2. 02ハイパーパラメーター調整
  3. 03複数指標による評価
  4. 04K-fold 検証
  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

コメント