BeastByteAI/scikit-llm
Seamlessly integrate LLMs into scikit-learn.
What it solves
大規模言語モデル (LLMs) をテキスト分析タスクに利用するプロセスを、scikit-learn エコシステムに直接統合することで簡素化します。これにより、ユーザーは使い慣れた scikit-learn API を離れることなく、ChatGPT などのモデルの力を活用して、ゼロショットテキスト分類などのタスクを実行できます。
How it works
このライブラリは、ZeroShotGPTClassifier のような scikit-learn と互換性のあるモデルを提供します。これらは標準的な scikit-learn のワークフローで使用可能です。ユーザーは SKLLMConfig を通じて API キーと組織 ID を設定し、fit と predict メソッドを使用することでテキスト分析を実行します。
Who it’s for
scikit-learn に精通しており、かつ LLM を既存のテキスト分析パイプラインに統合したいデータサイエンティストや機械学習エンジニアです。
Highlights
- Seamless integration with scikit-learn
- Support for zero-shot text classification using GPT models ** Support for zero-learn text classification using GPT models
- Simple configuration for OpenAI credentials
- Provides demo datasets for quick start