/ Модель

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

Контекст
512 токенов
Параметры
12 млн
Архитектура
bert · dense
Опубликована
2 мар 2022

bert-mini-finetuned-age_news-classification

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

Компактная англоязычная модель на базе BERT-Mini для классификации новостей по категориям AG News.

1 094скачиваний за 30 дней9лайковвеса обновлены20 мар 2023

Бенчмарки

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

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

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

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

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

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

mrm8488/bert-mini-finetuned-age_news-classification — англоязычная модель классификации новостей, созданная разработчиком mrm8488 дообучением BERT-Mini на наборе данных AG News. Это компактный классификатор последовательностей на архитектуре BERT с примерно 12 млн параметров и контекстом до 512 токенов. Модель принимает текст и присваивает ему категорию новости; в карточке Hugging Face для тестовой выборки AG News указана точность около 0,934.

Модель предназначена для автоматической тематической сортировки новостных материалов и демонстрации классификации текста. Карточка предоставляет варианты весов PyTorch и Flax, а также формат safetensors. Условия лицензии в доступных источниках не подтверждены.

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

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

Ограничения

Модель обучена для классификации новостей AG News и не предназначена для произвольных тематик без дополнительной адаптации. Контекст ограничен 512 токенами.

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

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

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

Также подходит для

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

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

Открыть на Hugging Face

BERT-Mini fine-tuned on age_news dataset for news classification

Test set accuray: 0.93

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей