CoolFace
Modelpublic

esmarquez17/fine-tunning-roberta-bne-hate-offensive

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

fine-tunning-roberta-bne-hate-offensive

This model is a fine-tuned version of BSC-LT/roberta-base-bne on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.6211
  • Hate/offensive F1: 0.8958
  • Hate/offensive Precision: 0.9421
  • Hate/offensive Recall: 0.8539
  • Normal F1: 0.9519
  • Normal Precision: 0.9308
  • Normal Recall: 0.9740
  • Micro F1: 0.9342
  • Acc: 0.9342
  • Macro F1: 0.9239
  • Macro Precision: 0.9364
  • Macro Recall: 0.9139

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 LossHate/offensive F1Hate/offensive PrecisionHate/offensive RecallNormal F1Normal PrecisionNormal RecallMicro F1AccMacro F1Macro PrecisionMacro Recall
0.28651.010640.33980.89530.93240.86100.95100.93360.96910.93320.93320.92310.93300.9150
0.15272.021280.34170.89420.92040.86950.94970.93700.96270.93180.93180.92200.92870.9161
0.0473.031920.56260.89140.96680.82700.95180.91990.98590.93320.93320.92160.94340.9064
0.01454.042560.58740.89860.93960.86100.95280.93380.97260.93560.93560.92570.93670.9168
0.00295.053200.62110.89580.94210.85390.95190.93080.97400.93420.93420.92390.93640.9139

Framework versions

  • Transformers 4.35.2
  • Pytorch 2.1.0+cu118
  • Datasets 2.15.0
  • Tokenizers 0.15.0