Happy Transformer makes it easy to fine-tune NLP Transformer models and use them for inference.
Documentation and news: happytransformer.com
Join our Discord server: 
Happy Transformer makes it easy to fine-tune NLP Transformer models and use them for inference.
Read about the full 3.0.0 update including breaking changes here.
| Tasks | Inference | Training |
|---|---|---|
| Text Generation | ✔ | ✔ |
| Text Classification | ✔ | ✔ |
| Word Prediction | ✔ | ✔ |
| Question Answering | ✔ | ✔ |
| Text-to-Text | ✔ | ✔ |
| Next Sentence Prediction | ✔ | |
| Token Classification | ✔ |
pip install happytransformer
from happytransformer import HappyWordPrediction
#--------------------------------------#
happy_wp = HappyWordPrediction() # default uses distilbert-base-uncased
result = happy_wp.predict_mask("I think therefore I [MASK]")
print(result) # [WordPredictionResult(token='am', score=0.10172799974679947)]
print(result[0].token) # am
Text generation with training (GPT-Neo)
Text classification (training)
Text classification (hate speech detection)
Text classification (sentiment analysis)
FAQs
Happy Transformer makes it easy to fine-tune NLP Transformer models and use them for inference.
We found that happytransformer demonstrated a healthy version release cadence and project activity because the last version was released less than a year ago. It has 1 open source maintainer collaborating on the project.
Did you know?
Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.
Security News
Research
The Socket Research Team breaks down a malicious wrapper package that uses obfuscation to harvest credentials and exfiltrate sensitive data.
Research
Security News
Attackers used a malicious npm package typosquatting a popular ESLint plugin to steal sensitive data, execute commands, and exploit developer systems.
Security News
The Ultralytics' PyPI Package was compromised four times in one weekend through GitHub Actions cache poisoning and failure to rotate previously compromised API tokens.