CoolFace
Modelpublic

elfsmo/XLM-Roberta-SDGs-Oplib-Elsevier-Pruned

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

XLM-Roberta-SDGs-Oplib-Elsevier-Pruned

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

  • —Loss: 0.1548
  • —Accuracy: 0.4267
  • —F1 Micro: 0.8408
  • —F1 Macro: 0.8114
  • —Precision Micro: 0.7984
  • —Precision Macro: 0.7779
  • —Recall Micro: 0.8880
  • —Recall Macro: 0.8513
  • —Roc Auc: 0.9206
  • —Hamming Loss: 0.0579

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: 1.1154214038433235e-05
  • —trainbatchsize: 4
  • —evalbatchsize: 4
  • —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_ratio: 0.1311822395684484
  • —num_epochs: 10/20
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossAccuracyF1 MicroF1 MacroPrecision MicroPrecision MacroRecall MicroRecall MacroRoc AucHamming Loss
0.33831.028600.29840.03890.62050.39180.52800.38950.75230.47960.80620.1585
0.21412.057200.20690.18930.74470.64230.65190.56380.86840.75630.88590.1026
0.17023.085800.16510.30930.79120.70530.72240.66560.87460.75740.90230.0795
0.14124.0114400.15270.35040.81160.76360.74780.71880.88740.82360.91250.0710
0.11765.0143000.14670.40370.82750.78990.77970.76200.88160.82770.91490.0633
0.10226.0171600.14480.43520.83880.80940.81820.80390.86050.82140.91040.0570
0.08467.0200200.14200.42560.83790.80530.79410.77190.88680.84430.91950.0591
0.06978.0228800.14330.45230.84630.81600.82110.80200.87310.83260.91670.0546
0.05929.0257400.14860.43310.84410.80970.81190.78360.87890.84330.91830.0559
0.049510.0286000.15480.42670.84080.81140.79840.77790.88800.85130.92060.0579

Framework versions

  • —Transformers 4.51.3
  • —Pytorch 2.7.0+cu126
  • —Datasets 3.6.0
  • —Tokenizers 0.21.1