PulseWave/INVOICE-DISPUTE
PulseWave
Clasificación de texto
Este es un modelo SetFit que se puede usar para la clasificación de texto. Se utiliza una instancia de LogisticRegression para la clasificación. El modelo ha sido entrenado utilizando una técnica de aprendizaje eficiente de pocos disparos que involucra: Afinar un Transformer de oración con aprendizaje contrastivo. Entrenar una cabeza de clasificación con características del Transformer de oración afinado.
Como usar
Uso directo para inferencia
Primero instale la biblioteca SetFit:
pip install setfit
Luego puede cargar este modelo y ejecutar la inferencia:
from setfit import SetFitModel
# Descargar desde el Hub de 🤗
model = SetFitModel.from_pretrained("setfit_model_id")
# Ejecutar inferencia
preds = model("¡Me encantó la película de Spiderman!")
Funcionalidades
- Tipo de modelo: SetFit
- Cabeza de clasificación: una instancia de LogisticRegression
- Longitud máxima de secuencia: 512 tokens
- Número de clases: 2 clases
Casos de uso
- Clasificación de texto