Textattack/bert-base-uncased-yelp-polarity

textattack
Clasificación de texto

Este modelo bert-base-uncased fue ajustado para la clasificación de secuencias utilizando TextAttack y el conjunto de datos yelp_polarity cargado mediante la biblioteca nlp. El modelo fue ajustado durante 5 épocas con un tamaño de lote de 16, una tasa de aprendizaje de 5e-05 y una longitud máxima de secuencia de 256. Dado que se trataba de una tarea de clasificación, el modelo fue entrenado con una función de pérdida de entropía cruzada. La mejor puntuación que el modelo consiguió en esta tarea fue 0.9699473684210527, medida por la precisión del conjunto de evaluación, encontrada después de 4 épocas.

Como usar

Este modelo bert-base-uncased fue ajustado para la clasificación de secuencias utilizando TextAttack y el conjunto de datos yelp_polarity cargado mediante la biblioteca nlp. El modelo fue ajustado durante 5 épocas con un tamaño de lote de 16, una tasa de aprendizaje de 5e-05 y una longitud máxima de secuencia de 256.

Para más información, consulta TextAttack en Github.

Funcionalidades

Clasificación de texto
Transformers
PyTorch
JAX
Puntos de Inferencia

Casos de uso

Recopilación de opiniones polarizadas de Yelp
Clasificación de sentimientos
Análisis de comentarios de clientes