happytransformer 2.2.2

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
Word Prediction	✔	✔
Text Generation	✔	✔
Text Classification	✔	✔
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](https://github.com/ericfillion)  Lead Maintainer
- [Ted Brownlow](https://github.com/ted537) Maintainer