BeastByteAI/scikit-llm

Seamlessly integrate LLMs into scikit-learn.

What it solves

它透過將大型語言模型 (LLMs) 直接整合至 scikit-learn 生態系統中，簡化了使用 LLMs 進行文本分析任務的流程。這讓使用者能夠在不離開熟悉的 scikit-learn API 的情況下，利用 ChatGPT 等模型的威力，執行如零樣本文本分類等任務。

How it works

此函式庫提供與 scikit-learn 相容的模型，例如 ZeroShotGPTClassifier，可與標準的 scikit-learn 工作流程配合使用。使用者透過 SKLLMConfig 設定其 API 金鑰與組織 ID，接著使用 fit 與 predict 方法來執行文本分析。

Who it’s for

對於已經熟悉 scikit-learn 並希望將 LLMs 整合至現有文本分析流程中的數據科學家與機器學習工程師。

Highlights

  • Seamless integration with scikit-learn
  • Support for zero-shot text classification using GPT models
  • Simple configuration for OpenAI credentials
  • Provides demo datasets for quick start