This is a repository of NLP models trained by NLP Town. If you need custom models, get in touch through our website.
Model name | Size | Description |
---|---|---|
en_sentiment_reviews_sm | 22MB | Sentiment classification trained on 3.6M Amazon reviews. |
Install the models using pip:
pip install MODEL_URL
You can consult the metadata of a model like this:
python -m spacy info en_sentiment_reviews_sm
>>> import spacy
>>> nlp = spacy.load("en_sentiment_reviews_sm")
>>> nlp("This is a horrible movie").cats
{'positive': 0.0016374080441892147, 'negative': 0.9983626008033752}
>>> nlp("This is a great movie").cats
{'positive': 0.995779275894165, 'negative': 0.004220753442496061}