Elron/bleurt-base-512

Elron

Clasificación de texto

Versión de PyTorch de los modelos originales de BLEURT del artículo ACL "BLEURT: Learning Robust Metrics for Text Generation" de Thibault Sellam, Dipanjan Das y Ankur P. Parikh de Google Research.

Como usar

from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch

tokenizer = AutoTokenizer.from_pretrained("Elron/bleurt-base-512")
model = AutoModelForSequenceClassification.from_pretrained("Elron/bleurt-base-512")
model.eval()

references = ["hola mundo", "hola mundo"]
candidates = ["hola universo", "adiós mundo"]

with torch.no_grad():
    scores = model(**tokenizer(references, candidates, return_tensors='pt'))[0].squeeze()

print(scores) # tensor([1.0327, 0.2055])

Funcionalidades

Clasificación de texto
Transformers
PyTorch
Puntos de inferencia

Casos de uso

Generación de texto robusta