CoolFace
Modelpublic

peter2000/distilbert-base-uncased-finetuned-osdg

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
0likes62downloads
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-finetuned-osdg

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

  • —Loss: 1.8193
  • —F1 Score: 0.7962
  • —Accuracy: 0.8434

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: 5e-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 LossF1 ScoreAccuracy
0.37691.010170.82580.77290.8257
0.27592.020340.83640.77730.8262
0.14123.030511.02030.78330.8379
0.14234.040681.16030.76830.8224
0.09395.050851.30290.78430.8329
0.07576.061021.35620.79310.8379
0.08017.071191.29250.78400.8395
0.03118.081361.46320.77500.8318
0.02639.091531.57600.78430.8312
0.019610.0101701.56890.78900.8417
0.031311.0111871.60340.79090.8417
0.000712.0122041.67250.78890.8406
0.008113.0132211.64630.79110.8395
0.006114.0142381.77300.78610.8345
0.00315.0152551.80010.78470.8379
0.000216.0162721.73280.79120.8434
0.017.0172891.79140.80110.8489
0.000918.0183061.77720.79580.8456
0.019.0193231.80280.79580.8434
0.020.0203401.81930.79620.8434

Framework versions

  • —Transformers 4.19.2
  • —Pytorch 1.11.0+cu113
  • —Datasets 2.2.2
  • —Tokenizers 0.12.1