/ Модель

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

Контекст
512 токенов
Параметры
138 млн
Лицензия
MIT
Архитектура
bert · dense
Опубликована
2 мар 2022

indobert-base-uncased-finetuned-indonlu-smsa

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

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

36скачиваний за 30 дней0лайковвеса обновлены2 фев 2024

Бенчмарки

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

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

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

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

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

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

ayameRushia/indobert-base-uncased-finetuned-indonlu-smsa — модель на базе BERT, дообученная на датасете IndoNLU для классификации текста на индонезийском языке. Она предназначена для анализа тональности и классификации коротких сообщений; в README приведены результаты на тестовой выборке: accuracy 0,9302 и F1 0,9066. Модель содержит 137,8 млн параметров, поддерживает контекст до 512 токенов и доступна в форматах PyTorch и Safetensors. Лицензия — MIT.

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

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

Ограничения

Модель ориентирована на индонезийский язык и классификацию текста; качество на других языках и задачах не подтверждено. Приведённые метрики получены на оценочной выборке IndoNLU и не гарантируют такое же качество на пользовательских данных.

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

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

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

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

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

Открыть на Hugging Face

indobert-base-uncased-finetuned-indonlu-smsa

This model is a fine-tuned version of indolem/indobert-base-uncased on the indonlu dataset. It achieves the following results on the evaluation set:

  • Loss: 0.2277
  • Accuracy: 0.9302
  • F1: 0.9066
  • Precision: 0.8992
  • Recall: 0.9147

Training hyperparameters

The following hyperparameters were used during training:

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

Training results

Training LossEpochStepValidation LossAccuracyF1PrecisionRecall
No log1.03440.38310.84760.77150.78170.7627
0.41672.06880.28090.89050.84060.86990.8185
0.26243.010320.22540.92300.88420.90040.8714
0.26244.013760.23780.92380.87970.91800.8594
0.18655.017200.22770.93020.90660.89920.9147
0.12176.020640.24440.92620.89810.90130.8957
0.12177.024080.29850.92860.89990.90350.8971
0.08478.027520.33970.92780.89690.90900.8871
0.05519.030960.35420.92700.89610.90100.8924
0.055110.034400.38620.92220.88950.89700.8846

Framework versions

  • Transformers 4.14.1
  • Pytorch 1.10.0+cu111
  • Datasets 1.17.0
  • Tokenizers 0.10.3

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей