/ Модель

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

Контекст
512 токенов
Параметры
146 млн
Архитектура
bert · dense

bert-fa-zwnj-base-ner

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

Персидская модель BERT для извлечения именованных сущностей, включая людей, организации, места, даты и суммы.

352скачиваний за 30 дней5лайковвеса обновлены18 мая 2021

Бенчмарки

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

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

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

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

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

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

HooshvareLab/bert-fa-zwnj-base-ner — модель BERT, дообученная для распознавания именованных сущностей на персидском языке. Она размечает текст по десяти типам сущностей: даты, события, объекты инфраструктуры, местоположения, суммы денег, организации, проценты, люди, продукты и время. Модель предназначена для извлечения таких сущностей из персидских текстов и запускается через Transformers pipeline для NER.

Модель содержит 145,5 млн параметров и обрабатывает последовательности длиной до 512 токенов. Для обучения использован смешанный набор данных ARMAN, PEYMA и WikiANN. В README приведена общая оценка F1 0,957 на тестовых данных; результаты зависят от типа сущности и распределения данных. Модель доступна в форматах PyTorch, TensorFlow и Flax. Лицензия в предоставленных данных не указана.

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

  • извлечение имён людей из персидских документов
  • поиск организаций и мест в персидских текстах
  • извлечение дат, времени и событий из персидских новостей
  • распознавание денежных сумм и процентов в персидских текстах
  • разметка упоминаний продуктов и объектов инфраструктуры

Ограничения

Модель обучена на смешанном наборе ARMAN, PEYMA и WikiANN; качество может снижаться на текстах и типах сущностей, отличающихся от обучающих данных. Контекст ограничен 512 токенами.

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

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

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

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

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

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

Открыть на Hugging Face

BertNER

This model fine-tuned for the Named Entity Recognition (NER) task on a mixed NER dataset collected from ARMAN, PEYMA, and WikiANN that covered ten types of entities:

  • Date (DAT)
  • Event (EVE)
  • Facility (FAC)
  • Location (LOC)
  • Money (MON)
  • Organization (ORG)
  • Percent (PCT)
  • Person (PER)
  • Product (PRO)
  • Time (TIM)

Dataset Information

RecordsB-DATB-EVEB-FACB-LOCB-MONB-ORGB-PCTB-PERB-PROB-TIMI-DATI-EVEI-FACI-LOCI-MONI-ORGI-PCTI-PERI-PROI-TIM
Train291331423148714001391941715926355123471855150194750182421411810591957957376991914332
Valid514226725325023621002651642173317193737993877172703260101138230335
Test60494072562482886983216942646318435688884088582633967141170729678

Evaluation

The following tables summarize the scores obtained by model overall and per each class.

Overall

Modelaccuracyprecisionrecallf1
Bert0.9950860.9534540.9611130.957268

Per entities

numberprecisionrecallf1
DAT4070.8606360.8648650.862745
EVE2560.9695820.9960940.982659
FAC2480.9761900.9919350.984000
LOC28840.9702320.9719140.971072
MON980.9052630.8775510.891192
ORG32160.9391250.9546020.946800
PCT941.0000000.9680850.983784
PER26450.9652440.9659740.965608
PRO3180.9814811.0000000.990654
TIM430.6923080.8372090.757895

How To Use

You use this model with Transformers pipeline for NER.

Installing requirements

pip install transformers

How to predict using pipeline

from transformers import AutoTokenizer
from transformers import AutoModelForTokenClassification  # for pytorch
from transformers import TFAutoModelForTokenClassification  # for tensorflow
from transformers import pipeline


model_name_or_path = "HooshvareLab/bert-fa-zwnj-base-ner" 
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
model = AutoModelForTokenClassification.from_pretrained(model_name_or_path)  # Pytorch
# model = TFAutoModelForTokenClassification.from_pretrained(model_name_or_path)  # Tensorflow

nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "در سال ۲۰۱۳ درگذشت و آندرتیکر و کین برای او مراسم یادبود گرفتند."

ner_results = nlp(example)
print(ner_results)

Questions?

Post a Github issue on the ParsNER Issues repo.

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей