moshew/bge-small-en-v1.5_setfit-sst2-english
moshew
Clasificación de texto
Este es un modelo SetFit que se puede usar para la clasificación de texto. El modelo ha sido entrenado utilizando una técnica eficiente de aprendizaje con pocos ejemplos que implica: - Entrenamiento fino de un Transformador de Oraciones ("BAAI/bge-small-en-v1.5") con aprendizaje contrastivo. - Entrenamiento de una cabeza de clasificación con características del Transformador de Oraciones afinado.
Como usar
Para usar este modelo para la inferencia, primero instala la biblioteca SetFit:
python -m pip install setfit
Luego puedes ejecutar la inferencia de la siguiente manera:
from setfit import SetFitModel
# Descargar desde Hub y ejecutar inferencia
model = SetFitModel.from_pretrained("moshew/bge-small-en-v1.5_setfit-sst2-english")
# Ejecutar inferencia
preds = model(["¡Me encantó la película de Spiderman!", "La piña en la pizza es lo peor 🤮"])
Funcionalidades
- Clasificación de texto
- Transformador de Oraciones
- Aprendizaje con pocos ejemplos
- Aprendizaje contrastivo
- Entrenamiento eficiente
Casos de uso
- Clasificación de sentimientos de textos
- Análisis de opiniones