metadata
language:
- es
tags:
- spanish
- sentiment
datasets:
- muchocine
electricidad-base-muchocine-finetuned
This model fine-tunes mrm8488/electricidad-base-discriminator on muchocine dataset for sentiment classification to predict star_rating.
How to use
The model can be used directly with the HuggingFace pipeline
.
from transformers import AutoTokenizer, AutoModelWithLMHead
tokenizer = AutoTokenizer.from_pretrained("shahp7575/gpt2-horoscopes")
model = AutoModelWithLMHead.from_pretrained("shahp7575/gpt2-horoscopes")
Examples
from transformers import pipeline
clf = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
clf('¡Qué película tan fantástica! ¡Me alegro de haberlo visto!')
>>> [{'label': '5', 'score': 0.9156607389450073}]
clf("La historia y el casting fueron geniales.")
>>> [{'label': '4', 'score': 0.6666394472122192}]
clf("Me gustó pero podría ser mejor.")
>>> [{'label': '3', 'score': 0.7013391852378845}]
clf("dinero tirado en esta pelicula")
>>> [{'label': '2', 'score': 0.7564149498939514}]