/ Модель

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

Контекст
512 токенов
Параметры
20,4 млн
Лицензия
CC BY 4.0
Архитектура
electra · dense
Опубликована
12 мар 2022

small-e-czech-finetuned-ner-wikiann

Разметка токеновТекст → label6 профессий

Компактная модель ELECTRA для распознавания именованных сущностей в чешских текстах.

13 088скачиваний за 30 дней3лайкавеса обновлены12 мар 2022

Бенчмарки

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

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

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

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

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

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

Модель richielo/small-e-czech-finetuned-ner-wikiann — компактная модель распознавания именованных сущностей для чешского языка, разработанная путём дообучения Seznam/small-e-czech на датасете WikiANN. Она основана на архитектуре ELECTRA, содержит 20,4 млн параметров и принимает последовательности длиной до 512 токенов. Модель присваивает метки отдельным токенам и подходит для извлечения сущностей из чешских текстов.

На оценочной выборке указаны precision 0,8713, recall 0,8970, F1 0,8840 и accuracy 0,9557. Модель распространяется в формате PyTorch по лицензии CC BY 4.0. Оценки относятся к указанной выборке; подробности о составе классов и поведении на других текстах в карточке не приведены.

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

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

Ограничения

Показатели качества приведены только для оценочной выборки WikiANN и могут отличаться на текстах другого типа. Карточка не описывает набор меток и подробные ограничения применения.

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

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

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

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

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

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

Открыть на Hugging Face

small-e-czech-finetuned-ner-wikiann

This model is a fine-tuned version of Seznam/small-e-czech on the wikiann dataset. It achieves the following results on the evaluation set:

  • Loss: 0.2547
  • Precision: 0.8713
  • Recall: 0.8970
  • F1: 0.8840
  • Accuracy: 0.9557

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: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 20

Training results

Training LossEpochStepValidation LossPrecisionRecallF1Accuracy
0.29241.025000.24490.76860.80880.78820.9320
0.20422.050000.21370.80500.83980.82200.9400
0.16993.075000.19120.82360.85930.84110.9466
0.14194.0100000.19310.83490.86710.85070.9488
0.13165.0125000.18920.84700.87760.86200.9519
0.10426.0150000.20580.84330.88110.86180.9508
0.08847.0175000.20200.86020.88490.87240.9531
0.09028.0200000.21180.85510.88370.86920.9528
0.06699.0225000.21710.86340.89060.87680.9550
0.052910.0250000.22280.86380.89120.87730.9545
0.061311.0275000.22930.86260.88980.87600.9544
0.054912.0300000.22760.86940.89580.88240.9554
0.051613.0325000.23840.87170.89400.88270.9552
0.041214.0350000.24430.87010.89310.88150.9554
0.034515.0375000.24640.87230.89580.88390.9557
0.041216.0400000.24770.87050.89480.88250.9552
0.036317.0425000.25250.87420.89730.88560.9559
0.034118.0450000.25290.87270.89620.88430.9561
0.019419.0475000.25330.86990.89660.88300.9557
0.024720.0500000.25470.87130.89700.88400.9557

Framework versions

  • Transformers 4.17.0
  • Pytorch 1.10.0+cu111
  • Datasets 1.18.4
  • Tokenizers 0.11.6

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей