HAPPY TRANSFORMER
Happy Transformer makes it easy to fine-tune and perform inference with NLP Transformer models.
Upcoming Version 4.0.0
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
| 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.
Install
pip install "happytransformer<4.0.0"
Maintainers
- Eric Fillion Lead Maintainer
- Ted Brownlow Maintainer
Tutorial Articles
Text classification (training)
Fine-tune a Grammar Correction Model
Tutorial Videos
Text generation with training (GPT-Neo)
Text classification (hate speech detection)