nikcheerla/amd-power-dialer-v1
nikcheerla
Clasificación de texto
Este es un modelo SetFit que se puede usar para clasificación de texto. El modelo ha sido entrenado utilizando una técnica eficiente de aprendizaje con pocos ejemplos que incluye: Afinar un Transformador de Oraciones con aprendizaje contrastivo y Entrenar una cabeza de clasificación con características del Transformador de Oraciones afinado.
Como usar
Para usar este modelo para inferencia, primero instale la biblioteca SetFit:
python -m pip install setfit
Luego puede ejecutar la inferencia de la siguiente manera:
from setfit import SetFitModel
# Descarga desde el Hub y ejecuta la inferencia
model = SetFitModel.from_pretrained("nikcheerla/amd-power-dialer-v1")
# Ejecuta la inferencia
preds = model(["¡Me encantó la película de Spiderman!", "La piña en la pizza es lo peor 🤮"])
Funcionalidades
- Afinación de un Transformador de Oraciones con aprendizaje contrastivo
- Entrenamiento de una cabeza de clasificación con características del Transformador de Oraciones afinado
- Compatibilidad con bibliotecas: sentence-transformers, PyTorch, SetFit
Casos de uso
- Clasificación de sentimientos
- Clasificación de opiniones
- Identificación de temas en textos