/ Модель

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

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

categorization-finetuned-20220721-164940-pruned-20220803-184651

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

Дообученный классификатор на базе RoBERTa для присвоения категорий текстовым последовательностям.

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

Бенчмарки

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

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

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

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

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

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

Модель carted-nlp/categorization-finetuned-20220721-164940-pruned-20220803-184651 — дообученный классификатор текста на базе RoBERTa, разработанный carted-nlp. Она предназначена для категоризации текстовых последовательностей и относится к архитектуре RobertaForSequenceClassification; содержит около 120,9 млн параметров и принимает последовательности длиной до 514 токенов.

В карточке приведены результаты оценки: accuracy 0,8760 и F1 0,8751 при loss 0,4673. Датасет указан как None, а назначение категорий, языки и прикладная область не раскрыты. Модель опубликована в формате PyTorch; сведения о лицензии в предоставленных данных отсутствуют.

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

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

Ограничения

Категории, обучающий датасет и поддерживаемые языки не описаны, поэтому пригодность для конкретной области или языка не подтверждена. Контекст ограничен 514 токенами.

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

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

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

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

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

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

Открыть на Hugging Face

categorization-finetuned-20220721-164940-pruned-20220803-184651

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.4673
  • Accuracy: 0.8760
  • F1: 0.8751

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: 7e-06
  • train_batch_size: 48
  • eval_batch_size: 48
  • seed: 314
  • gradient_accumulation_steps: 6
  • total_train_batch_size: 288
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • lr_scheduler_warmup_steps: 1000
  • num_epochs: 15

Training results

Training LossEpochStepValidation LossAccuracyF1
0.34040.5120000.43290.88720.8865
0.34331.0140000.42800.88830.8876
0.32811.5260000.43020.88900.8883
0.3312.0280000.42650.88910.8885
0.32242.53100000.43000.88810.8874
0.33613.04120000.42910.88890.8882
0.33233.54140000.43370.88780.8871
0.35564.05160000.43450.88570.8851
0.36634.56180000.44170.88360.8828
0.39025.06200000.45550.87890.8781
0.40365.57220000.45560.87880.8779
0.43056.07240000.46970.87510.8742
0.45016.58260000.47630.87380.8725
0.47337.09280000.48570.87100.8700
0.48517.59300000.48630.87050.8695
0.48468.1320000.48490.87080.8698
0.48568.61340000.48350.87070.8695
0.47749.11360000.47970.87190.8708
0.46359.62380000.47760.87280.8717
0.456110.12400000.47460.87390.8729
0.447510.63420000.47050.87490.8740
0.441311.14440000.46910.87540.8744
0.438911.64460000.46790.87600.8750
0.436112.15480000.46770.87590.8749
0.436212.65500000.46720.87630.8753
0.430913.16520000.46710.87610.8751
0.431613.67540000.46700.87640.8754
0.432114.17560000.46680.87640.8755
0.431114.68580000.46680.87640.8754

Framework versions

  • Transformers 4.18.0.dev0
  • Pytorch 1.9.1+cu111
  • Datasets 2.3.2
  • Tokenizers 0.11.6

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей