CoolFace
Modelpublic

christinacdl/clickbait_binary_detection_DeBERTa

sourceHugging Facemitupdated 3y agoView on Hugging Face
0likes19downloads
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. -->

clickbaitbinarydetection_DeBERTa

This model is a fine-tuned version of microsoft/deberta-v3-large on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.7269
  • Macro F1: 0.9010
  • Micro F1: 0.9069
  • Accuracy: 0.9069

Performance on test set:

  • Accuracy: 0.911986301369863
  • F1 score: 0.9053903329555788
  • Precision: 0.9069346899004087
  • Recall : 0.9039394560612273
  • Matthews Correlation Coefficient: 0.8108686139956713
  • Precision of each class: [0.92560647 0.88826291]
  • Recall of each class: [0.93518519 0.87269373]
  • F1 score of each class: [0.93037117 0.88040949]

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-06
  • trainbatchsize: 4
  • evalbatchsize: 8
  • seed: 42
  • gradientaccumulationsteps: 2
  • totaltrainbatch_size: 8
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • num_epochs: 10
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossMacro F1Micro F1Accuracy
0.26921.054750.26760.90510.91420.9142
0.24922.0109510.33310.90780.91560.9156
0.21893.0164260.39090.91070.91690.9169
0.17694.0219020.37990.91140.91780.9178
0.14795.0273770.51030.89800.90320.9032
0.1086.0328530.52150.91230.91830.9183
0.09577.0383280.65490.89740.90280.9028
0.07738.0438040.67680.90440.91010.9101
0.05869.0492790.68370.90230.90830.9083
0.043910.0547500.72690.90100.90690.9069

Framework versions

  • Transformers 4.27.1
  • Pytorch 2.0.1+cu118
  • Datasets 2.9.0
  • Tokenizers 0.13.3