/ Модель

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

Контекст
514 токенов
Параметры
129 млн
Лицензия
CC BY 4.0
Архитектура
roberta · dense
Базовая модель
BantuBERTa
Библиотека
transformers
Опубликована
29 янв 2025

BantuBERTa-vmw-finetuned

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

Дообученная на базе BantuBERTa модель RoBERTa для классификации текстов.

13скачиваний за 30 дней0лайковвеса обновлены29 янв 2025

Бенчмарки

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

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

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

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

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

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

Kuongan/BantuBERTa-vmw-finetuned — дообученная модель классификации текстов на основе dsfsi/BantuBERTa, разработанная Kuongan. Она использует архитектуру RoBERTa и предназначена для задач классификации последовательностей. Веса представлены в формате Safetensors; модель содержит 129,3 млн параметров и поддерживает контекст до 514 токенов. Базовая BantuBERTa адаптирована для NLP-задач на языках банту, однако язык и конкретная предметная область этого дообучения не описаны.

В README указаны результаты оценки: loss 0,3269, F1 0,1633, ROC AUC 0,5507 и accuracy 0,4729. Эти метрики указывают на ограниченное качество в оценённой задаче; назначение меток и состав данных не раскрыты. Модель распространяется по лицензии CC BY 4.0.

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

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

Ограничения

F1 0,1633 и accuracy 0,4729 на указанном оценочном наборе свидетельствуют об ограниченном качестве; состав данных и значение классов не описаны. Контекст ограничен 514 токенами.

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

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

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

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

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

Открыть на Hugging Face

BantuBERTa-vmw-finetuned

This model is a fine-tuned version of dsfsi/BantuBERTa on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.3269
  • F1: 0.1633
  • Roc Auc: 0.5507
  • Accuracy: 0.4729

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: 32
  • eval_batch_size: 32
  • seed: 42
  • optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: cosine
  • lr_scheduler_warmup_steps: 100
  • num_epochs: 20

Training results

Training LossEpochStepValidation LossF1Roc AucAccuracy
0.40951.0660.31120.00.50.4612
0.29082.01320.29030.00.50.4612
0.2843.01980.28820.00.50.4612
0.29584.02640.28790.00.50.4612
0.27495.03300.28510.01390.50330.4651
0.2656.03960.28680.05740.51730.4729
0.22817.04620.28020.10730.53380.4845
0.19638.05280.28760.11090.53530.4651
0.17499.05940.30400.09660.52820.4496
0.153410.06600.30460.13220.53960.4612
0.133611.07260.31290.16020.54730.4690
0.119412.07920.32220.12870.53720.4612
0.110613.08580.32690.16330.55070.4729
0.100414.09240.32960.12690.53600.4496
0.09115.09900.33640.16120.55130.4690
0.085516.010560.33470.15700.54720.4690
0.07917.011220.33990.15130.54400.4574

Framework versions

  • Transformers 4.47.0
  • Pytorch 2.5.1+cu121
  • Datasets 3.2.0
  • Tokenizers 0.21.0

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей