README.md

metadata

language: es
license: mit
tags:
  - roberta
  - text-classification
  - spanish
  - sentiment-analysis
pipeline_tag: text-classification

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)