kasparas12/crypto_individual_infer_model_setfit
kasparas12
Clasificación de texto
Este es un modelo SetFit que se puede usar para clasificación de texto. Este modelo SetFit utiliza BAAI/bge-small-en-v1.5 como el modelo de incrustación de Sentence Transformer. Se utiliza una instancia de LogisticRegression para la clasificación. El modelo ha sido entrenado usando una técnica eficiente de aprendizaje con pocos ejemplos que incluye: - Ajuste fino de un Sentence Transformer con aprendizaje contrastivo. - Entrenamiento de una cabeza de clasificación con características del Sentence Transformer ajustado.
Como usar
Primero instala la librería SetFit:
pip install setfit
Luego puedes cargar este modelo y ejecutar la inferencia:
from setfit import SetFitModel
# Descargar desde el Hub de 🤗
model = SetFitModel.from_pretrained("kasparas12/crypto_individual_infer_model_setfit")
# Ejecutar inferencia
preds = model("productor negocios y en otros lugares en licencia mis propios puntos de vista la gran manzana")
Funcionalidades
- Tipo de Modelo: SetFit
- Cuerpo de Sentence Transformer: BAAI/bge-small-en-v1.5
- Cabeza de clasificación: una instancia de LogisticRegression
- Longitud máxima de secuencia: 512 tokens
- Número de clases: 28 clases
Casos de uso
- Clasificación de texto
- Inferencia directa para aplicaciones de clasificación de texto usando un transformador de oraciones preentrenado
- Uso eficiente del aprendizaje con pocos ejemplos en diversos contextos de clasificación