CoolFace
Modelpublic

saiteki-kai/QA-DeBERTa-v3-large-focal-binary

sourceHugging Facemitupdated 10mo agoView on Hugging Face
0likes49downloads
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. -->

QA-DeBERTa-v3-large-focal-binary

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

  • —Loss: 0.0614
  • —Accuracy: 0.8630
  • —Unsafe Precision: 0.8774
  • —Unsafe Recall: 0.8763
  • —Unsafe F1: 0.8768
  • —Unsafe Fpr: 0.1537
  • —Unsafe Aucpr: 0.9565
  • —Safe Precision: 0.8450
  • —Safe Recall: 0.8463
  • —Safe F1: 0.8457
  • —Safe Fpr: 0.1237
  • —Safe Aucpr: 0.9246

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: 6e-06
  • —trainbatchsize: 64
  • —evalbatchsize: 512
  • —seed: 42
  • —optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 1000
  • —num_epochs: 10

Training results

Training LossEpochStepValidation LossAccuracyUnsafe PrecisionUnsafe RecallUnsafe F1Unsafe FprUnsafe AucprSafe PrecisionSafe RecallSafe F1Safe FprSafe Aucpr
0.06960.250121140.06750.85050.88440.84140.86230.13800.94550.81240.86200.83650.15860.9023
0.05820.500142280.06520.85540.86130.88210.87160.17810.95150.84740.82190.83440.11790.9129
0.0660.750263420.06450.85960.89130.85160.87100.13040.95280.82370.86960.84600.14840.9159
0.06521.000284560.06220.85950.86520.88560.87520.17320.95410.85210.82680.83930.11440.9188
0.05961.2503105700.06470.86020.88850.85620.87210.13480.95440.82740.86520.84590.14380.9196
0.05991.5004126840.06180.85990.87650.87110.87380.15400.95430.83950.84600.84270.12890.9203
0.05581.7504147980.06070.86230.88350.86690.87510.14350.95600.83680.85650.84660.13310.9234
0.05632.0005169120.06140.86300.87730.87630.87680.15370.95650.84500.84630.84570.12370.9246
0.05132.2505190260.06290.86090.89060.85500.87250.13170.95560.82680.86830.84700.14500.9224
0.05172.5006211400.06410.85910.90180.83800.86880.11440.95550.81340.88560.84790.16200.9232
0.05322.7507232540.06490.86110.90030.84390.87120.11720.95630.81840.88280.84940.15610.9243
0.05553.0007253680.06300.86220.88100.86980.87540.14750.95650.83920.85250.84580.13020.9253

Framework versions

  • —Transformers 4.57.3
  • —Pytorch 2.7.1+cu118
  • —Datasets 4.4.1
  • —Tokenizers 0.22.1