Modelo de Polaridad SetFit con BAAI/bge-small-en-v1.5 en SemEval 2014 Task 4 (Restaurantes)

tomaarsen
Clasificación de texto

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