Sultannn's picture
Update README.md
770bbf1
metadata
license: apache-2.0
tags:
  - generated_from_keras_callback
model-index:
  - name: Sultannn/bert-base-ft-ner-xtreme-id
    results: []
language: id
datasets:
  - xtreme

Sultannn/bert-base-ft-ner-xtreme-id

This model is a fine-tuned version of bert-base-multilingual-uncased on an xtreme PAN-X.id for NER downstream task.

Details of the downstream task (NER) - Dataset

Dataset # Examples
Train 35 K
Validation 5 K

Metrics on evaluation set

Metrics Score
Accuracy 97.18
F1 93.26
Precision 92.36
Recall 94.18

Training hyperparameters

The following hyperparameters were used during training:

  • optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 4e-05, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 4e-05, 'decay_steps': 8248, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'passive_serialization': True}, 'warmup_steps': 500, 'power': 1.0, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
  • training_precision: mixed_float16

Framework versions