/ Модель

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

Контекст
512 токенов
Параметры
136 млн
Лицензия
Apache 2.0
Архитектура
bert · dense
Опубликована
16 июн 2022

bert-finetuned-ner

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

Дообученная BERT-модель для распознавания и классификации токенов, включая упоминания белковых мутаций, SNP и мутаций ДНК.

27скачиваний за 30 дней0лайковвеса обновлены28 июн 2022

Бенчмарки

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

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

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

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

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

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

Salvatore разработал BERT-модель для классификации токенов, дообученную на основе bert-base-cased. Она предназначена для распознавания сущностей и меток на уровне токенов; приведённые результаты оценки указывают на классы Proteinmutation, Snp и Dnamutation. Модель содержит 135,5 млн параметров, использует формат PyTorch и принимает последовательности длиной до 512 токенов.

В README указаны метрики на оценочном наборе: F1 0,2398 и accuracy 0,9703; по классам F1 составляет 0,1309 для Proteinmutation, 0,1953 для Snp и 0,3778 для Dnamutation. Общие сведения о назначении датасета и ограничениях автор не приводит. Модель распространяется по лицензии Apache-2.0.

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

  • поиск упоминаний мутаций ДНК в текстах
  • извлечение упоминаний SNP из документов
  • распознавание упоминаний белковых мутаций
  • разметка токенов для анализа биомедицинских текстов

Ограничения

Общий F1 на оценочном наборе составляет 0,2398, а F1 для классов белковых мутаций и SNP — 0,1309 и 0,1953 соответственно. Название и состав обучающего датасета в README не раскрыты.

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

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

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

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

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

Открыть на Hugging Face

bert-finetuned-ner

This model is a fine-tuned version of bert-base-cased on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0997
  • Proteinmutation F1: 0.1309
  • Snp F1: 0.1953
  • Dnamutation F1: 0.3778
  • Precision: 0.2380
  • Recall: 0.2416
  • F1: 0.2398
  • Accuracy: 0.9703

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

Training results

Training LossEpochStepValidation LossProteinmutation F1Snp F1Dnamutation F1PrecisionRecallF1Accuracy
No log1.03240.05330.03960.28300.46670.23340.32210.27070.9788
0.10722.06480.04370.60650.49060.50090.48020.63480.54680.9868
0.10723.09720.05920.13790.24850.20050.16390.22280.18890.9731
0.05734.012960.07220.07490.25300.46920.27050.29590.28260.9749
0.04315.016200.07660.15740.18470.25400.17660.22850.19920.9723
0.04316.019440.08050.10990.22020.23830.16570.20970.18510.9715
0.03967.022680.08860.13370.21380.43180.26830.26780.26800.9724
0.03548.025920.09270.15350.21130.37690.25050.25280.25160.9714
0.03549.029160.09780.10110.25400.38120.24950.25280.25120.9705
0.031210.032400.09970.13090.19530.37780.23800.24160.23980.9703

Framework versions

  • Transformers 4.17.0
  • Pytorch 1.10.2
  • Datasets 2.0.0
  • Tokenizers 0.12.1

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей