happytransformer 2.2.3

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.

Happy Transformer

Documentation and news: happytransformer.com

Join our brand new Discord server:

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	✔	✔
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 classification (hate speech detection)

Text classification (sentiment analysis)

Text generation with training (GPT-Neo)

Word prediction with training (DistilBERT, RoBERTa)