Files
transformers/model_cards/mrm8488/bert-spanish-cased-finetuned-ner/README.md
T
2020-02-17 10:58:43 -05:00

2.0 KiB

language, thumbnail
language thumbnail
spanish https://i.imgur.com/jgBdimh.png

Spanish BERT (BETO) + NER

This model is a fine-tuned on NER-C of the Spanish BERT cased (BETO) for NER downstream task.

Details of the downstream task (NER) - Dataset

I preprocessed the dataset and splitted it as train / dev (80/20)

Dataset # Examples
Train 8.7 K
Dev 2.2 K
!export NER_DIR='/content/ner_dataset'
!python /content/transformers/examples/run_ner.py \
  --model_type bert \
  --model_name_or_path dccuchile/bert-base-spanish-wwm-cased \
  --do_train \
  --do_eval \
  --data_dir '/content/ner_dataset' \
  --num_train_epochs 15.0 \
  --max_seq_length 384 \
  --output_dir /content/model_output \
  --save_steps 5000 \

Comparison:

Model # score
bert-base-spanish-wwm-cased (BETO) 88.43
bert-spanish-cased-finetuned-ner (this one) 89.65
Best Multilingual BERT 87.38
 ***** All metrics on Eval results  *****

f1 = 0.8965040489828165
loss = 0.11504213575173258
precision = 0.893679858239811
recall = 0.8993461462254805

Created by Manuel Romero/@mrm8488

Made with ♥ in Spain