Scikit-LLM: Scikit-Learn 与大语言模型的结合
无缝集成像 ChatGPT 这样的强大语言模型到 scikit-learn 中,以增强文本分析任务。
安装 💾
pip install scikit-llm
支持我们 🤝
您可以通过以下方式支持该项目:
- ⭐ 在 GitHub 上为 Scikit-LLM 点星(点击右上角的星星按钮)
- 💡 在 issues 部分或 Discord 提供您的反馈或提出建议
- 📰 在 LinkedIn 或其他平台上发布关于 Scikit-LLM 的文章
- 🔗 查看我们的其他项目:Dingo,Falcon
快速开始和文档 📚
使用 GPT 进行零样本文本分类的快速启动示例:
# 导入必要模块
from skllm.datasets import get_classification_dataset
from skllm.config import SKLLMConfig
from skllm.models.gpt.classification.zero_shot import ZeroShotGPTClassifier
# 配置凭证
SKLLMConfig.set_openai_key("<YOUR_KEY>")
SKLLMConfig.set_openai_org("<YOUR_ORGANIZATION_ID>")
# 加载演示数据集
X, y = get_classification_dataset() # 标签:positive, negative, neutral
# 初始化模型并进行预测
clf = ZeroShotGPTClassifier(model="gpt-4")
clf.fit(X,y)
clf.predict(X)
更多信息请参考 文档。
引用
您可以使用以下 BibTeX 来引用 Scikit-LLM:
@software{ScikitLLM,
author = {Iryna Kondrashchenko and Oleh Kostromin},
year = {2023},
publisher = {beastbyte.ai},
address = {Linz, Austria},
title = {Scikit-LLM: Scikit-Learn Meets Large Language Models},
url = {https://github.com/iryna-kondr/scikit-llm }
}