/ Модель

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

Контекст
512 токенов
Параметры
137 млн
Архитектура
bert · dense

bert-base-uncased-yelp-polarity

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

Дообученная модель BERT определяет положительную или отрицательную тональность отзывов Yelp на английском языке.

23 256скачиваний за 30 дней5лайковвеса обновлены20 мая 2021

Бенчмарки

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

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

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

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

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

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

Модель textattack/bert-base-uncased-yelp-polarity — версия BERT base uncased, дообученная TextAttack для классификации тональности отзывов Yelp на положительные и отрицательные. Она обучена на наборе yelp_polarity в течение пяти эпох; наилучшая точность на проверочной выборке составила около 97%. Модель содержит 136,7 млн параметров, использует архитектуру BertForSequenceClassification и принимает текстовые последовательности длиной до 512 токенов.

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

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

  • Классификация отзывов Yelp по тональности
  • Определение положительных и отрицательных отзывов
  • Автоматическая сортировка отзывов клиентов по полярности

Ограничения

Модель обучена на отзывах Yelp, поэтому качество на текстах других жанров и предметных областях может отличаться. При обучении использовалась максимальная длина 256 токенов; оценка точности относится к проверочной выборке исходной задачи.

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

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

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

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

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

Открыть на Hugging Face

TextAttack Model Card

This bert-base-uncased model was fine-tuned for sequence classification using TextAttack and the yelp_polarity dataset loaded using the nlp library. The model was fine-tuned for 5 epochs with a batch size of 16, a learning rate of 5e-05, and a maximum sequence length of 256. Since this was a classification task, the model was trained with a cross-entropy loss function. The best score the model achieved on this task was 0.9699473684210527, as measured by the eval set accuracy, found after 4 epochs.

For more information, check out TextAttack on Github.

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей