CoolFace
Modelpublic

Svetlana0303/Regression_xlnet_aug_CustomLoss

sourceHugging Facemitupdated 3y agoView on Hugging Face
0likes20downloads
Model Card

<!-- This model card has been generated automatically according to the information Keras had access to. You should probably proofread and complete it, then remove this comment. -->

Regressionxlnetaug_CustomLoss

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

  • —Train Loss: 0.2430
  • —Train Mae: 0.5316
  • —Train Mse: 0.4353
  • —Train R2-score: 0.4207
  • —Validation Loss: 0.2455
  • —Validation Mae: 0.5751
  • —Validation Mse: 0.4288
  • —Validation R2-score: 0.6784
  • —Epoch: 14

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:

  • —optimizer: {'name': 'Adam', 'weightdecay': None, 'clipnorm': None, 'globalclipnorm': None, 'clipvalue': None, 'useema': False, 'emamomentum': 0.99, 'emaoverwritefrequency': None, 'jitcompile': True, 'islegacyoptimizer': False, 'learningrate': 1e-04, 'beta1': 0.9, 'beta2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
  • —training_precision: float32

Training results

Train LossTrain MaeTrain MseTrain R2-scoreValidation LossValidation MaeValidation MseValidation R2-scoreEpoch
0.29500.57890.48960.69090.25120.53410.48010.76030
0.26590.55160.45380.71450.28280.56800.52820.74771
0.26560.54920.45870.68580.23370.53450.44120.74312
0.25630.54840.44900.72470.24130.52020.46190.75813
0.25890.55110.45420.67570.24110.51990.46150.75804
0.25370.54070.44370.76050.23590.52440.44950.75175
0.24940.53850.43990.76680.25100.58210.43010.66216
0.24950.54030.44240.77650.23600.52420.44960.75197
0.25010.53940.43830.52090.23490.52790.44640.74918
0.24460.53430.43460.75340.23660.55850.42980.71059
0.24390.53160.43230.75610.25430.53760.48530.759910
0.24150.53480.43300.79280.23410.53160.44340.745911
0.24080.53230.42890.78270.23460.52910.44540.748112
0.24990.53920.44100.60080.23640.52300.45080.752713
0.24300.53160.43530.42070.24550.57510.42880.678414

Framework versions

  • —Transformers 4.28.1
  • —TensorFlow 2.12.0
  • —Datasets 2.12.0
  • —Tokenizers 0.13.3