tomaarsen/setfit-absa-bge-small-en-v1.5-restaurants-aspect
Este es un modelo SetFit entrenado en el conjunto de datos SemEval 2014 Task 4 (Restaurantes) que se puede usar 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. Una instancia de LogisticRegression se usa para la clasificación. En particular, este modelo se encarga de filtrar los candidatos de intervalo de aspectos. El modelo ha sido entrenado utilizando una técnica eficiente de aprendizaje con pocos ejemplos que implica: - Ajustar un Sentence Transformer con aprendizaje contrastivo. - Entrenar una cabeza de clasificación con características del Sentence Transformer ajustado. Este modelo fue entrenado dentro del contexto de un sistema más grande para ABSA, que se ve así: 1. Usar un modelo spaCy para seleccionar posibles candidatos de intervalo de aspectos. 2. Usar este modelo SetFit para filtrar estos posibles candidatos de intervalo de aspectos. 3. Usar un modelo SetFit para clasificar los candidatos de intervalo de aspectos filtrados.
Como usar
Primero instale la biblioteca SetFit:
pip install setfit
Luego puede cargar este modelo y ejecutar la inferencia.
from setfit import AbsaModel
# Descargar desde el Hub de 🤗
model = AbsaModel.from_pretrained(
"tomaarsen/setfit-absa-bge-small-en-v1.5-restaurants-aspect",
"tomaarsen/setfit-absa-bge-small-en-v1.5-restaurants-polarity",
)
# Ejecutar inferencia
preds = model("La comida estaba excelente, pero el lugar era demasiado concurrido.")
Funcionalidades
- Modelo SetFit
- Cuerpo de Sentence Transformer: BAAI/bge-small-en-v1.5
- Cabeza de clasificación: instancia de LogisticRegression
- Modelo spaCy: en_core_web_lg
- Modelo de Aspectos SetFitABSA: tomaarsen/setfit-absa-bge-small-en-v1.5-restaurants-aspect
- Modelo de Polaridad SetFitABSA: tomaarsen/setfit-absa-bge-small-en-v1.5-restaurants-polarity
- Longitud máxima de secuencia: 512 tokens
- Número de clases: 2 clases
- Conjunto de datos de entrenamiento: SemEval 2014 Task 4 (Restaurantes)
- Licencia: apache-2.0
Casos de uso
- Análisis de Sentimientos Basado en Aspectos (ABSA)
- Filtración de candidatos de intervalo de aspectos en reseñas de restaurantes
- Clasificación de candidatos de intervalo de aspectos en reseñas de restaurantes