CoolFace
Modelpublic

VampeeHuntee/bert-base-multilingual-cased_baseline_words

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes5downloads
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-base-multilingual-casedbaselinewords

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

  • —Loss: 0.0998
  • —Patient Id: 0.9840
  • —Name: 0.9182
  • —Gender: 0.9623
  • —Age: 0.9725
  • —Job: 0.7799
  • —Location: 0.9501
  • —Organization: 0.8965
  • —Date: 0.9869
  • —Symptom And Disease: 0.8626
  • —Transportation: 0.9885
  • —F1 Macro: 0.9302
  • —F1 Micro: 0.9466

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
  • —trainbatchsize: 8
  • —evalbatchsize: 8
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 5

Training results

Training LossEpochStepValidation LossPatient IdNameGenderAgeJobLocationOrganizationDateSymptom And DiseaseTransportationF1 MacroF1 Micro
0.19961.06290.12070.97280.89970.84140.90180.50200.90300.74960.98470.80640.90530.84670.8959
0.05842.012580.09780.97930.91010.94760.97260.50440.93800.87380.98600.84330.90910.88640.9319
0.03443.018870.09250.97760.91250.94550.97660.72160.94170.86650.98650.86340.96550.91570.9383
0.02224.025160.09710.98360.91780.95560.97390.74420.94930.88830.98650.85790.99430.92510.9443
0.01475.031450.09980.98400.91820.96230.97250.77990.95010.89650.98690.86260.98850.93020.9466

Framework versions

  • —Transformers 4.41.2
  • —Pytorch 2.1.2
  • —Datasets 2.19.2
  • —Tokenizers 0.19.1