fnnx-ai/scikit-llm

What it solves

它簡化了將 ChatGPT 等大型語言模型 (LLMs) 整合進熟悉的 scikit-learn 生態系統以進行文本分析任務的過程，讓使用者能在標準的機器學習 pipeline 中利用 LLM 的能力。

How it works

此函式庫為 LLM 提供與 scikit-learn 相容的封裝器 (wrappers)。例如，它提供了一個 ZeroShotGPTClassifier，讓使用者透過設定 OpenAI API key 並使用 scikit-learn 模型中常見的標準 .fit() 和 .predict() 方法來執行零樣本 (zero-shot) 文本分類。

Who it's for

對於已經熟悉 scikit-learn 並希望在不重寫現有工作流程的情況下使用 LLM 進行文本分析的資料科學家和機器學習工程師。

Highlights

• 與 scikit-learn 的 API 無縫整合
• 支援使用 GPT 模型進行零樣本文本分類
• 透過 SKLLMConfig 進行簡易配置