Happy Transformer makes it easy to fine-tune and perform inference with NLP Transformer models.
Version 4.0.0 is in the final stages of development. It's a complete rewrite of the library with many new features. It includes breaking changes, and so we recommend you set the version to <4.0.0 in your requirements.
| Tasks | Inference | Training |
|---|---|---|
| Text Generation | ✔ | ✔ |
| Text Classification | ✔ | ✔ |
| Word Prediction | ✔ | ✔ |
| Question Answering | ✔ | ✔ |
| Text-to-Text | ✔ | ✔ |
| Next Sentence Prediction | ✔ | |
| Token Classification | ✔ |
Note: word prediction, question answering, next sentence prediction and token classification have been deprecated.
pip install "happytransformer<4.0.0"- Eric Fillion Lead Maintainer
- Ted Brownlow Maintainer
Text classification (training)
Fine-tune a Grammar Correction Model
Text generation with training (GPT-Neo)
Text classification (hate speech detection)