/ Модель

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

Контекст
514 токенов
Параметры
36 млн
Архитектура
roberta · dense
Опубликована
11 авг 2022

categorization-finetuned-20220721-164940-distilled-20220811-074207

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

Компактная модель RoBERTa для классификации и категоризации текстов.

12скачиваний за 30 дней0лайковвеса обновлены11 авг 2022

Бенчмарки

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

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

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

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

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

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

Модель carted-nlp/categorization-finetuned-20220721-164940-distilled-20220811-074207 — дообученная Carted NLP модель RoBERTa для классификации текста. Она содержит 36 млн параметров и принимает последовательности длиной до 514 токенов. В README указаны метрики на оценочном наборе: accuracy 0,8771 и F1 0,8763; название предполагает применение для категоризации, но конкретные категории и обучающие данные не описаны.

Модель подходит для присвоения текстам категорий в прикладных системах. Доступен формат PyTorch. Лицензия в предоставленных данных не указана.

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

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

Ограничения

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

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

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

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

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

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

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

Открыть на Hugging Face

categorization-finetuned-20220721-164940-distilled-20220811-074207

This model is a fine-tuned version of carted-nlp/categorization-finetuned-20220721-164940 on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.1499
  • Accuracy: 0.8771
  • F1: 0.8763

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: 2e-05
  • train_batch_size: 64
  • eval_batch_size: 96
  • seed: 314
  • gradient_accumulation_steps: 4
  • total_train_batch_size: 256
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • lr_scheduler_warmup_steps: 1500
  • num_epochs: 30.0

Training results

Training LossEpochStepValidation LossAccuracyF1
0.56440.5625000.27390.78220.7774
0.26581.1250000.22880.81590.8127
0.23071.6975000.20820.82980.8273
0.21262.25100000.19700.83890.8370
0.20122.81125000.18880.84500.8433
0.19033.37150000.18290.84960.8485
0.18463.94175000.17830.85290.8511
0.17714.5200000.17500.85480.8537
0.17265.06225000.17270.85770.8564
0.16735.62250000.16830.86020.8591
0.16486.19275000.16750.86080.8597
0.15966.75300000.16570.86300.8620
0.15637.31325000.16350.86460.8639
0.1547.87350000.16130.86560.8647
0.14968.43375000.16110.86660.8656
0.14969.0400000.15980.86760.8669
0.14459.56425000.15940.86810.8671
0.143510.12450000.15880.86880.8679
0.140710.68475000.15680.87030.8695
0.138211.25500000.15640.87080.8700
0.137211.81525000.15500.87200.8713
0.134412.37550000.15590.87180.8708
0.133712.93575000.15400.87350.8729
0.130313.5600000.15410.87290.8721
0.130414.06625000.15310.87350.8727
0.127414.62650000.15350.87360.8727
0.126615.18675000.15270.87500.8742
0.125115.74700000.15250.87550.8748
0.123416.31725000.15280.87530.8745
0.122916.87750000.15160.87600.8753
0.12117.43775000.15230.87590.8752
0.121217.99800000.15150.87600.8754
0.118518.56825000.15140.87650.8757
0.118619.12850000.15160.87660.8760
0.117219.68875000.15060.87740.8767
0.116420.24900000.15130.87700.8763
0.11620.81925000.15070.87740.8767
0.114521.37950000.15070.87770.8770
0.114321.93975000.15060.87760.8770
0.113122.491000000.15070.87790.8772
0.113123.051025000.15050.87790.8772
0.112323.621050000.15060.87810.8774
0.111724.181075000.15040.87830.8776
0.111824.741100000.15030.87840.8777
0.111125.31125000.15030.87830.8776
0.111125.871150000.15020.87840.8777
0.110526.431175000.15040.87830.8776
0.110526.991200000.15020.87860.8779
0.110427.551225000.15030.87860.8779
0.109628.121250000.15020.87850.8779
0.110128.681275000.15010.87860.8779
0.110129.241300000.15020.87860.8779
0.109429.81325000.15010.87860.8779

Framework versions

  • Transformers 4.17.0
  • Pytorch 1.11.0+cu113
  • Datasets 2.3.2
  • Tokenizers 0.11.6

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей