Views
No views yet
cardiffnlp/twitter-xlm-roberta-base-sentiment| Métrica | Valor |
|---|---|
| Accuracy | 0.7580 |
| F1 Score | 0.7386 |
| Precision | 0.7344 |
| Recall | 0.7580 |
1from transformers import XLMRobertaForSequenceClassification, XLMRobertaTokenizer
2import torch
3
4# Cargar el modelo y el tokenizador
5model_path = "natmarinn/sentimientos-bullrich"
6model = XLMRobertaForSequenceClassification.from_pretrained(model_path)
7tokenizer = XLMRobertaTokenizer.from_pretrained(model_path)
8
9# Texto de ejemplo
10texto = "Vamos pato"
11
12# Tokenización
13inputs = tokenizer(texto, return_tensors="pt", truncation=True)
14
15# Predicción
16with torch.no_grad():
17 outputs = model(**inputs)
18 logits = outputs.logits
19 pred_class = torch.argmax(logits, dim=1).item()
20
21# Mostrar resultado
22clases = ["Clase 0", "Clase 1", "Clase 2"]
23print(f"El comentario es clasificado como: {clases[pred_class]}")