Happy Transformer is an API built on top of Hugging Face's Transformer library that makes it easy to utilize state-of-the-art NLP models.
Project description
Happy Transformer
Documentation and news: happytransformer.com
New Course: Create a text generation web app. Also learn how to fine-tune GPT-Neo link
Happy Transformer is an package built on top of Hugging Face's transformer library that makes it easy to utilize state-of-the-art NLP models.
Features
Public Methods | Basic Usage | Training |
---|---|---|
Text Generation | ✔ | ✔ |
Text Classification | ✔ | ✔ |
Word Prediction | ✔ | ✔ |
Question Answering | ✔ | ✔ |
Text-to-Text | ✔ | ✔ |
Next Sentence Prediction | ✔ | |
Token Classification | ✔ |
Quick Start
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
Maintainers
- Eric Fillion Lead Maintainer
- Ted Brownlow Maintainer
Tutorials
Text generation with training (GPT-Neo)
Text classification (training)
Text classification (hate speech detection)
Text classification (sentiment analysis)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
happytransformer-2.3.3.tar.gz
(25.6 kB
view hashes)
Built Distribution
Close
Hashes for happytransformer-2.3.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 485b8c28dd83d9e90427ac6ae52384be80c4871ae2d42eb3b65fa6aa0be1b69f |
|
MD5 | 930548d83f386aa7408ecdebcd9dd1ac |
|
BLAKE2b-256 | f5cae0c6e82c1ab0417fa673e54d31dfa909908a4fe0da5116782a3a7e4db56e |