Scikit-LLM: Scikit-Learn Meets Large Language Models
Seamlessly integrate powerful language models like ChatGPT into scikit-learn for enhanced text analysis tasks.
Installation 💾
pip install scikit-llm
Support us 🤝
You can support the project in the following ways:
- ⭐ Star Scikit-LLM on GitHub (click the star button in the top right corner)
- 💡 Provide your feedback or propose ideas in the issues section or Discord
- 📰 Post about Scikit-LLM on LinkedIn or other platforms
- 🔗 Check out our other projects: Dingo, Falcon
Quick Start & Documentation 📚
Quick start example of zero-shot text classification using GPT:
# Import the necessary modules
from skllm.datasets import get_classification_dataset
from skllm.config import SKLLMConfig
from skllm.models.gpt.classification.zero_shot import ZeroShotGPTClassifier
# Configure the credentials
SKLLMConfig.set_openai_key("<YOUR_KEY>")
SKLLMConfig.set_openai_org("<YOUR_ORGANIZATION_ID>")
# Load a demo dataset
X, y = get_classification_dataset() # labels: positive, negative, neutral
# Initialize the model and make the predictions
clf = ZeroShotGPTClassifier(model="gpt-4")
clf.fit(X,y)
clf.predict(X)
For more information please refer to the documentation.
Citation
You can cite Scikit-LLM using the following BibTeX:
@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 }
}
1 maintainer
Metadata
1 maintainer
Release files for scikit-llm 1.4.3
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| scikit_llm-1.4.3.tar.gz | 38.5 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| scikit_llm-1.4.3-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 97.9 kB
Release files / scikit_llm-1.4.3.tar.gz
| Download URL | scikit_llm-1.4.3.tar.gz |
|---|---|
| Size | 38.5 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
bc7d4965134179330fa3930c2d376dc64b6846b1fa4014f82dacd1eba46bd189
|
|
BLAKE2b-256 checksum How to use checksums |
85ee9799566b3cb21b4138dccb4c0c49dfa2252544515a1c1a443f007736ea50
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.10.19
|
Release files / scikit_llm-1.4.3-py3-none-any.whl
| Download URL | scikit_llm-1.4.3-py3-none-any.whl |
|---|---|
| Size | 59.4 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
7b5dfab5872c78dceb6641b37deb15b06b098fd7cd23ba60bbc152de6bb5d0b2
|
|
BLAKE2b-256 checksum How to use checksums |
6988de878d02b31a4a1572ffac215701036c297ca78a3068913a98e5dc472a96
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.10.19
|
