PROPUESTA/PROPIEDAD-DE-CUENTA
PulseWave
Clasificación de texto
Este es un modelo SetFit que se puede usar para Clasificación de Texto. Una instancia de LogisticRegression se usa para la clasificación. El modelo ha sido entrenado usando una técnica eficiente de aprendizaje con pocos ejemplos que implica: Ajuste fino de un Transformador de Oraciones con aprendizaje contrastivo. Entrenamiento de una cabeza de clasificación con características del Transformador de Oraciones ajustado.
Como usar
Instalar la biblioteca SetFit:
pip install setfit
Luego puede cargar este modelo y ejecutar inferencia.
from setfit import SetFitModel
# Descargar desde el 🤗 Hub
model = SetFitModel.from_pretrained("setfit_model_id")
# Ejecutar inferencia
preds = model("¡Me encantó la película de Spiderman!")
Funcionalidades
- Modelo SetFit
- Cabeza de clasificación: instancia de LogisticRegression
- Longitud máxima de secuencia: 512 tokens
- Número de clases: 2 clases
Casos de uso
- Uso directo para inferencia