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 and predict 方法进行文本分析。

Who it’s for

对于已经熟悉 scikit-learn 并希望将 LLMs 集成到现有文本分析流水线中的数据科学家和机器学习工程师。

Highlights

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