CoolFace
Modelpublic

dmargutierrez/distilbert-base-uncased-mapa-ner-coarse_grained-v2

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
0likes9downloads
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. -->

distilbert-base-uncased-mapa-ner-coarse_grained-v2

This model is a fine-tuned version of distilbert-base-uncased on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.1627
  • Precision: 0.7898
  • Recall: 0.4843
  • F1: 0.6004
  • Accuracy: 0.9857

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

Training results

Training LossEpochStepValidation LossPrecisionRecallF1Accuracy
0.03541.017390.08340.70230.46120.55680.9840
0.02552.034780.10340.81720.43550.56820.9852
0.01683.052170.09690.77140.45880.57540.9848
0.01324.069560.10420.74770.48380.58750.9852
0.01125.086950.11090.74210.48630.58760.9849
0.00856.0104340.10760.71940.49510.58650.9850
0.00677.0121730.13430.78280.45870.57840.9849
0.00478.0139120.12520.74250.48400.58600.9853
0.00459.0156510.14100.79430.46150.58380.9852
0.003510.0173900.13110.76240.49290.59870.9857
0.00311.0191290.14940.80590.46910.59300.9855
0.002512.0208680.14360.76740.48520.59450.9856
0.00213.0226070.15130.77780.47410.58910.9852
0.001414.0243460.15770.79860.47260.59380.9855
0.001615.0260850.15730.78020.47660.59170.9855
0.001116.0278240.15990.79170.47230.59160.9856
0.001217.0295630.16010.78480.48670.60080.9857
0.00118.0313020.15720.76140.49390.59910.9856
0.001119.0330410.16020.78580.48700.60130.9857
0.000920.0347800.16270.78980.48430.60040.9857

Framework versions

  • Transformers 4.26.0
  • Pytorch 1.13.1+cu117
  • Datasets 2.9.0
  • Tokenizers 0.13.2