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 进行轻松配置