/ Модель

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

Контекст
512 токенов
Параметры
136 млн
Лицензия
Apache 2.0
Архитектура
bert · dense
Базовая модель
bert-base-cased
Опубликована
15 апр 2024

mitre-bert-base-cased

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

Дообученная модель BERT для классификации текстовых последовательностей.

20скачиваний за 30 дней1лайквеса обновлены15 апр 2024

Бенчмарки

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

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

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

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

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

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

mitre-bert-base-cased — модель классификации последовательностей на архитектуре BERT, дообученная автором репозитория на наборе данных, обозначенном как None. Она предназначена для классификации текста. Модель содержит 136,1 млн параметров, принимает текстовые последовательности длиной до 512 токенов и распространяется в формате Safetensors.

В карточке приведены результаты оценки: accuracy 0,6994 и loss 1,0145. Назначение классов, состав обучающих данных и подробные ограничения не описаны. Модель основана на google-bert/bert-base-cased и лицензирована по Apache-2.0.

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

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

Ограничения

Названия классов и обучающий набор данных не раскрыты; на оценочной выборке указана точность 0,6994. Контекст ограничен 512 токенами.

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

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

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

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

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

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

Открыть на Hugging Face

mitre-bert-base-cased

This model is a fine-tuned version of bert-base-cased on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 1.0145
  • Accuracy: 0.6994

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05
  • train_batch_size: 16
  • eval_batch_size: 16
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 10

Training results

Training LossEpochStepValidation LossAccuracy
1.27610.685000.84530.6864
0.74481.3610000.75660.7164
0.60562.0415000.71870.7318
0.47632.7220000.71340.7307
0.42763.4125000.76040.7420
0.38554.0930000.74930.7362
0.33034.7735000.77270.7423
0.3135.4540000.80530.7263
0.29486.1345000.85550.7280
0.27796.8150000.88390.7127
0.25267.4955000.90970.7144
0.25768.1760000.94210.7171
0.24618.8665000.98210.7018
0.23579.5470001.01450.6994

Framework versions

  • Transformers 4.38.2
  • Pytorch 2.2.1+cu121
  • Datasets 2.18.0
  • Tokenizers 0.15.2

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей