/ Модель

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

Контекст
512 токенов
Параметры
137 млн
Лицензия
Apache 2.0
Архитектура
bert · dense
Опубликована
2 мар 2022

bert-base-uncased-sst2

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

Дообученная на SST-2 англоязычная BERT-модель для классификации тональности текста.

704скачиваний за 30 дней0лайковвеса обновлены10 фев 2025

Бенчмарки

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

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

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

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

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

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

Модель Yoshitomo Matsubara — версия BERT-base-uncased, дообученная на датасете SST-2 для классификации тональности коротких текстов на английском языке. Она использует архитектуру BERT с 136,7 млн параметров и обрабатывает последовательности длиной до 512 токенов. В README обучение описано как воспроизводимый пример с использованием torchdistill и библиотек Hugging Face; указанный общий результат на тестах GLUE — 77,9 балла.

Модель подходит для определения положительной или отрицательной тональности английских фраз и отзывов. Она опубликована в форматах PyTorch и Safetensors под лицензией Apache-2.0. Обучение на SST-2 ограничивает специализацию модели анализом тональности; README не приводит отдельной оценки качества именно для этой модели на SST-2.

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

  • определение тональности коротких английских отзывов
  • классификация английских фраз как положительных или отрицательных
  • анализ тональности пользовательских комментариев на английском

Ограничения

Модель обучена на SST-2 и ориентирована на классификацию тональности английских текстов; качество на других языках, доменах и задачах не подтверждено приведёнными сведениями. README сообщает общий результат GLUE 77,9, но не приводит отдельную оценку на SST-2.

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

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

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

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

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

Открыть на Hugging Face

bert-base-uncased fine-tuned on SST-2 dataset, using torchdistill and Google Colab.
The hyperparameters are the same as those in Hugging Face's example and/or the paper of BERT, and the training configuration (including hyperparameters) is available here.
I submitted prediction files to the GLUE leaderboard, and the overall GLUE score was 77.9.

Yoshitomo Matsubara: "torchdistill Meets Hugging Face Libraries for Reproducible, Coding-Free Deep Learning Studies: A Case Study on NLP" at EMNLP 2023 Workshop for Natural Language Processing Open Source Software (NLP-OSS)

[Paper] [OpenReview] [Preprint]

@inproceedings{matsubara2023torchdistill,
  title={{torchdistill Meets Hugging Face Libraries for Reproducible, Coding-Free Deep Learning Studies: A Case Study on NLP}},
  author={Matsubara, Yoshitomo},
  booktitle={Proceedings of the 3rd Workshop for Natural Language Processing Open Source Software (NLP-OSS 2023)},
  publisher={Empirical Methods in Natural Language Processing},
  pages={153--164},
  year={2023}
}

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей