UnitHr/dstilh-maria

MarIA fine-tuned for Sentiment Classification (DSTILH)

Este modelo está basado en MarIA (RoBERTa-base en español) y ha sido fine-tuneado para realizar clasificación de sentimientos en español.

Detalles del modelo

• Arquitectura: RobertaForSequenceClassification
• Número de clases: 3
  • LABEL_0: negativo
  • LABEL_1: neutro
  • LABEL_2: positivo

Uso

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("UnitHr/dstilh-maria")
model = AutoModelForSequenceClassification.from_pretrained("UnitHr/dstilh-maria")

inputs = tokenizer("Este medicamento me sentó fatal", return_tensors="pt")
outputs = model(**inputs)