thegenerativegeneration/stay_or_go_conversation_classifier_s_v2
thegenerativegeneration
Clasificación de texto
Este es un modelo SetFit que se puede usar para la clasificación de texto. Este modelo SetFit utiliza intfloat/multilingual-e5-small como el modelo de incrustaciones de transformadores de oraciones. Se utiliza una instancia de SetFitHead para la clasificación. El modelo ha sido entrenado utilizando una técnica eficiente de aprendizaje con pocos ejemplos que involucra: - Ajuste fino de un transformador de oraciones con aprendizaje contrastivo. - Entrenamiento de una cabeza de clasificación con características del transformador de oraciones afinado.
Como usar
Para utilizar este modelo directamente 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("query: Tôi xin lỗi nhưng tôi phải đi")
Funcionalidades
- Modelo SetFit
- Cuerpo del transformador de oraciones: intfloat/multilingual-e5-small
- Cabeza de clasificación: una instancia de SetFitHead
- Longitud máxima de secuencia: 512 tokens
- Número de clases: 2 clases
Casos de uso
- Clasificación de texto
- Aprendizaje con pocos ejemplos sin invitaciones