fnnx-ai/scikit-llm

What it solves

ChatGPT などの大規模言語モデル (LLMs) を、テキスト分析タスクのための使い慣れた scikit-learn エコシステムに統合することを簡素化します。これにより、ユーザーは標準的な機械学習 pipeline を使用して LLM の能力を活用できるようになります。

How it works

このライブラリは、LLM 用の scikit-learn 互換のラッパーを提供します。例えば、 ZeroShotGPTClassifier を提供しており、ユーザーは OpenAI API key を設定し、scikit-learn モデルに共通する標準的な .fit() と .predict() メソッドを使用することで、ゼロショット (zero-shot) テキスト分類を実行できます。

Who it's for

scikit-learn に慣れており、既存のワークフローをワークフローを書き換えることなく LLM をテキスト分析に使用したいデータサイエンティストや機械学習エンジニアです。

Highlights

• scikit-learn の API とシームレスに統合
• GPT モデルを使用したゼロショット テキスト分類のサポート
• SKLLMConfig を介した簡単な設定