CoolFace
Modelpublic

christinacdl/clickbait_binary_detection

sourceHugging Facemitupdated 3y agoView on Hugging Face
2likes16downloads
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

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

  • Loss: 0.4630
  • Macro F1: 0.9155
  • Micro F1: 0.9215
  • Accuracy: 0.9215

Performance on test set:

  • Accuracy: 0.9257990867579908
  • F1 score: 0.9199282431058413
  • Precision: 0.9233793490724882
  • Recall : 0.9168756883647268
  • Matthews Correlation Coefficient: 0.8402298675576902
  • Precision of each class: [0.931899 0.91485969]
  • Recall of each class: [0.95152505 0.88222632]
  • F1 score of each class: [0.94160977 0.89824671]

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: 6
  • evalbatchsize: 10
  • seed: 42
  • gradientaccumulationsteps: 2
  • totaltrainbatch_size: 12
  • 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.22961.036500.22360.91050.91830.9183
0.2282.073010.27080.91150.91920.9192
0.20753.0109510.31410.91640.92240.9224
0.18814.0146020.32110.91430.92010.9201
0.185.0182520.38520.91300.91880.9188
0.18186.0219030.37840.91100.91740.9174
0.14957.0255530.46060.91060.91560.9156
0.14538.0292040.46300.91550.92150.9215

Framework versions

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