mrm8488/distilroberta-finetuned-banking77

mrm8488
Clasificación de texto

distilroberta-base afinado en el conjunto de datos banking77 para la clasificación de intenciones. Precisión en el conjunto de prueba: 0.896.

Como usar

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

ckpt = 'mrm8488/distilroberta-finetuned-banking77'
tokenizer = AutoTokenizer.from_pretrained(ckpt)
model = AutoModelForSequenceClassification.from_pretrained(ckpt)

classifier = pipeline('text-classification', tokenizer=tokenizer, model=model)
classifier('What is the base of the exchange rates?')
# Output: [{'label': 'exchange_rate', 'score': 0.8509947657585144}]

Funcionalidades

Clasificación de texto
Basado en Transformers
Utiliza el conjunto de datos banking77
Compatibilidad con PyTorch
Compatibilidad con Safetensors
Compatible con AutoTrain
Compatible con Puntos de Inferencia

Casos de uso

Clasificación de intenciones en el sector bancario
Análisis de consultas y solicitudes de clientes
Mejora del servicio al cliente en aplicaciones bancarias