tomaarsen/setfit-absa-bge-small-en-v1.5-restaurants-aspect

tomaarsen
Clasificación de texto

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