/ Модель

Характеристики

Контекст
512 токенов
Параметры
137 млн
Архитектура
bert · dense
Опубликована
17 мая 2024

NLP-Sentiment-Analysis-Airline-Tweets-with-BERT-V2

Классификация текстаТекст → label3 профессии

Модель на основе BERT определяет положительную, нейтральную или отрицательную тональность твитов об авиакомпаниях.

10скачиваний за 30 дней1лайквеса обновлены24 мая 2024

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

Mastika разработала модель классификации тональности сообщений об авиакомпаниях в Twitter на основе BERT. Это дообученная модель BERT для классификации текста по трём категориям: отрицательная, нейтральная и положительная тональность. Архитектура — BertForSequenceClassification, размер — 136,7 млн параметров, максимальная длина контекста — 512 токенов; веса опубликованы в формате safetensors.

Модель подходит для анализа тональности отзывов и сообщений пассажиров об авиакомпаниях. В описанных испытаниях точность составила около 82%; F1 для отрицательной тональности — 0,89, нейтральной — 0,62, положительной — 0,76. Результаты по нейтральным сообщениям заметно слабее, а автор отмечает колебания качества между эпохами и риск переобучения. В README лицензия не указана.

Сценарии применения

  • Определение тональности отзывов пассажиров об авиакомпаниях
  • Поиск негативных сообщений о перелётах в Twitter
  • Мониторинг удовлетворённости клиентов авиакомпании
  • Разметка твитов об авиакомпаниях по тональности

Ограничения

Качество на нейтральных сообщениях ниже, чем на отрицательных и положительных; автор также отмечает колебания качества между эпохами и возможное переобучение. В README лицензия не указана.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

#NLP-Sentiment-Analysis-Airline-Tweets-with-BERT-V2

This repository features sentiment analysis projects that leverage BERT, a leading NLP model. This project involves pre-processing, tokenization, and BERT customization for airline tweet sentiment classification. The tasks in this model use the original model "BERT base model (no casing)", uses a data set: https://www.kaggle.com/datasets/crowdflower/twitter-airline-sentiment, and there are several stages in achieving results, below are the evaluation sets Accuracy: 0.8203551912568307

Colab notebook for improvements: https://colab.research.google.com/drive/1IQen2iNXkjOgdzjyi7PQyLFqHyqHTF3A?usp=sharing

#classification report for more detailed evaluation :

precisionrecallf1-scoresupport
negative0.880.900.89959
neutral0.680.580.62293
positive0.720.810.76212
-------------------------------------------------
accuracy0.821464
macro avg0.760.760.761464
weighted avg0.820.820.821464

The sentiment classification model achieved a promising overall accuracy of 82.04%, built on BertForSequenceClassifi- cation and trained for 10 epochs using AdamW optimization. The model exhibited stable performance, with validation ac- curacy consistently between 0.79 to 0.81, indicating effective learning. Additionally, it showed high precision, particularly for negative sentiment (0.88), along with moderate scores for neutral (0.68) and positive (0.72) sentiments. These results were supported by recall and F1-score metrics, providing a comprehensive understanding of performance across sentiment classes. The analysis of the confusion matrix revealed strong alignment between model predictions and actual labels, al- beit with opportunities for improvement, such as addressing overfitting or parameter adjustment, evident from performance fluctuations across epochs.

Developed by:Mastika

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · text-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
BertForSequenceClassification
Ревизия весов
edc708a

0 моделей