CoolFace
Modelpublic

Kuongan/CS221-xlnet-base-cased-finetuned-augmentation

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes15downloads
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-xlnet-base-cased-finetuned-augmentation

This model is a fine-tuned version of xlnet-base-cased on the None dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.4526
  • —F1: 0.7317
  • —Roc Auc: 0.7987
  • —Accuracy: 0.4657

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.55661.0970.50830.35170.58910.2220
0.43512.01940.40280.66180.73400.3953
0.35723.02910.37410.67450.76090.4440
0.28844.03880.37780.69070.75910.4422
0.23425.04850.39550.71110.77610.4711
0.18656.05820.40250.72220.78470.4747
0.14167.06790.45460.69910.76860.4422
0.09428.07760.45260.73170.79870.4657
0.08219.08730.47730.71740.78860.4458
0.06410.09700.50170.71940.78560.4495
0.049811.010670.53120.71310.78120.4404

Framework versions

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