Modelo de Polaridad SetFit con BAAI/bge-small-en-v1.5 en SemEval 2014 Task 4 (Restaurantes)
Este es un modelo SetFit entrenado en el conjunto de datos SemEval 2014 Task 4 (Restaurantes) que se puede utilizar para el Análisis de Sentimientos Basado en Aspectos (ABSA). 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. En particular, este modelo se encarga de clasificar las polaridades de los aspectos. El modelo ha sido entrenado usando una técnica eficiente de aprendizaje con pocas muestras que implica: - Afinar un Transformer de Sentencias con aprendizaje contrastivo. - Entrenar una cabeza de clasificación con características del Transformer de Sentencias afinado. Este modelo fue entrenado dentro del contexto de un sistema más grande para ABSA, que se ve así: - Usar un modelo spaCy para seleccionar posibles candidatos a span de aspecto. - Usar un modelo SetFit para filtrar estos posibles candidatos a span de aspecto. - Usar este modelo SetFit para clasificar los candidatos a span de aspecto filtrados.
Como usar
Primero instala la biblioteca SetFit:
pip install setfit
Luego, puedes cargar este modelo y ejecutar inferencias.
from setfit import AbsaModel
# Descargar desde el Hub de 🤗
model = AbsaModel.from_pretrained(
"tomaarsen/setfit-absa-bge-small-en-v1.5-restaurantes-aspect",
"tomaarsen/setfit-absa-bge-small-en-v1.5-restaurantes-polarity",
)
# Ejecutar inferencias
preds = model("The food was great, but the venue is just way too busy.")
Funcionalidades
- Modelo SetFit
- Cuerpo de Transformer de Sentencias: BAAI/bge-small-en-v1.5
- Cabeza de clasificación: instancia de LogisticRegression
- Modelo spaCy: en_core_web_lg
- Modelo de Aspecto SetFitABSA: tomaarsen/setfit-absa-bge-small-en-v1.5-restaurantes-aspect
- Modelo de Polaridad SetFitABSA: tomaarsen/setfit-absa-bge-small-en-v1.5-restaurantes-polarity
- Longitud Máxima de Secuencia: 512 tokens
- Número de Clases: 4 clases
- Conjunto de Datos de Entrenamiento: SemEval 2014 Task 4 (Restaurantes)
- Idioma: en
- Licencia: apache-2.0
Casos de uso
- Análisis de Sentimientos Basado en Aspectos (ABSA)
- Clasificación de polaridades de aspectos
- Filtración de candidatos a span de aspecto
- Clasificación de candidatos a span de aspecto filtrados