CoolFace
Modelpublic

Kuongan/CS221-toxic-bert-finetuned-semeval

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

CS221-toxic-bert-finetuned-semeval

This model is a fine-tuned version of unitary/toxic-bert on the None dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.4539
  • —F1: 0.7204
  • —Roc Auc: 0.7908
  • —Accuracy: 0.3953

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: 32
  • —evalbatchsize: 32
  • —seed: 42
  • —optimizer: Use adamwtorch with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • —lrschedulertype: cosine
  • —lrschedulerwarmup_steps: 100
  • —num_epochs: 20

Training results

Training LossEpochStepValidation LossF1Roc AucAccuracy
0.57021.0700.56150.42180.61900.1552
0.46012.01400.45990.59930.70980.3177
0.36883.02100.41400.69300.76970.3718
0.27764.02800.39740.71630.78680.4007
0.225.03500.40020.71780.78920.3827
0.17856.04200.41800.71030.78170.4043
0.14217.04900.42630.71190.78350.3971
0.09198.05600.45390.72040.79080.3953
0.0839.06300.45790.71920.78930.3953
0.060210.07000.48600.71120.78290.3989
0.044611.07700.49310.70640.77930.3791

Framework versions

  • —Transformers 4.47.1
  • —Pytorch 2.5.1+cu121
  • —Datasets 3.2.0
  • —Tokenizers 0.21.0