kasparas12/es_modelo_organizacional
kasparas12
Clasificación de texto
Este es un modelo SetFit que se puede usar para Clasificación de Texto. Este modelo SetFit usa BAAI/bge-small-en-v1.5 como el modelo de incrustación de Transformador de Oraciones. Se utiliza una instancia de Regresión Logística para la clasificación. El modelo ha sido entrenado usando una técnica eficiente de aprendizaje con pocos ejemplos que implica: 1. Afinar un Transformador de Oraciones con aprendizaje contrastivo. 2. Entrenar una cabeza de clasificación con características del Transformador de Oraciones afinado.
Como usar
Para uso directo para Inferencia:
Primero instala la biblioteca SetFit:
pip install setfit
Luego puedes cargar este modelo y ejecutar inferencia:
from setfit import SetFitModel
# Descargar desde el Hub de 🤗
model = SetFitModel.from_pretrained("kasparas12/is_organizational_model")
# Ejecutar inferencia
preds = model(["tradermayne Mayne "])
Funcionalidades
- Tipo de Modelo: SetFit
- Cuerpo del Transformador de Oraciones: BAAI/bge-small-en-v1.5
- Cabeza de Clasificación: una instancia de Regresión Logística
- Longitud Máxima de Secuencia: 512 tokens
- Número de Clases: 2 clases
Casos de uso
- Clasificación de texto