Ricardouchub/xlmr-sentiment-es-en

XLM-R Sentiment EN/ES (Movie Reviews)

Clasificador binario (Positive/Negative) para reseñas de películas en inglés y español, fine-tuned desde xlm-roberta-base con Rotten Tomatoes movies and critic reviews dataset from Kaggle

Métricas:

Acc 0.8519 · F1 0.8876 · Prec 0.8646 · Rec 0.9119 · AUC 0.9260
Umbral recomendado:* *0.48

Uso rápido

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
m = "Ricardouchub/xlmr-sentiment-es-en"; thr = 0.48
tok = AutoTokenizer.from_pretrained(m, use_fast=True)
mdl = AutoModelForSequenceClassification.from_pretrained(m).eval()
enc = tok(["Excelente actuación, final predecible."], truncation=True, max_length=224, padding=True, return_tensors="pt")
p = torch.softmax(mdl(**enc).logits, dim=-1)[:,1].item()
print(("POSITIVE" if p>=thr else "NEGATIVE"), round(p*100,1), "%")

Notas: split por película (evita fuga); limpieza mínima de texto. No apto para usos sensibles.

Autor: Ricardo Urdaneta