CoolFace
Modelpublic

importsmart/bert-to-distilbert-NER

sourceHugging Facemitupdated 4y agoView on Hugging Face
0likes18downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

bert-to-distilbert-NER

This model is a fine-tuned version of dslim/bert-base-NER on the conll2003 dataset. It achieves the following results on the evaluation set:

  • —Loss: 44.0386
  • —Precision: 0.0145
  • —Recall: 0.0185
  • —F1: 0.0163
  • —Accuracy: 0.7597

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: 6e-05
  • —trainbatchsize: 128
  • —evalbatchsize: 128
  • —seed: 33
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 15
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossPrecisionRecallF1Accuracy
201.40121.0110133.72310.01530.01060.01250.7539
106.93172.022099.36290.02660.03050.02840.7593
81.36013.033080.37630.01590.02140.01830.7604
63.83254.044067.76200.01790.02440.02070.7599
52.02715.055059.08060.02030.02680.02310.7598
44.44196.066055.32080.02110.02780.02400.7603
39.23517.077052.45100.01700.02220.01930.7598
35.34388.088050.45760.02050.02680.02320.7604
32.73859.099048.34180.01730.02270.01970.7595
30.653110.0110046.73040.01470.01880.01650.7600
29.081111.0121046.33860.01510.01900.01680.7599
27.950112.0132045.45160.01630.02040.01810.7604
26.745213.0143044.34250.01540.01990.01730.7592
25.536714.0154044.04150.01460.01900.01650.7594
24.550715.0165044.03860.01450.01850.01630.7597

Framework versions

  • —Transformers 4.19.1
  • —Pytorch 1.11.0+cu113
  • —Datasets 2.2.1
  • —Tokenizers 0.12.1