roberta-base-stsb

WillHeld

Clasificación de texto

Este modelo es una versión ajustada de roberta-base en el conjunto de datos GLUE STSB. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 0.4221, Pearson: 0.9116, Spearmanr: 0.9092, Puntuación combinada: 0.9104.

Como usar

El siguiente código ejemplifica cómo usar este modelo:

from transformers import AutoModelForSequenceClassification, AutoTokenizer

modelo = AutoModelForSequenceClassification.from_pretrained('WillHeld/roberta-base-stsb')
tokenizador = AutoTokenizer.from_pretrained('WillHeld/roberta-base-stsb')

texto = "I like you. I love you"
inputs = tokenizador(texto, return_tensors='pt')
outputs = modelo(**inputs)

Funcionalidades

Clasificación de texto
Basado en el modelo roberta-base
Utiliza el conjunto de datos GLUE STSB

Casos de uso

Clasificación de textos
Evaluación de similitudes semánticas en textos
Tareas de análisis de sentimientos