CoolFace
Modelpublic

PrabalAryal/multiclassclassification

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

multiclassclassification

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

  • Train Loss: 0.2109
  • Train Accuracy: 0.9640
  • Validation Loss: 0.1246
  • Validation Accuracy: 0.9791
  • Epoch: 6

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': 'AdamWeightDecay', 'learningrate': 3e-05, 'decay': 0.0, 'beta1': 0.9, 'beta2': 0.999, 'epsilon': 1e-07, 'amsgrad': False, 'weightdecay_rate': 0.0}
  • training_precision: float32

Training results

Train LossTrain AccuracyValidation LossValidation AccuracyEpoch
4.31080.14973.31910.28570
2.72190.44401.79100.67641
1.52520.71850.88250.85782
0.80690.86900.43190.93543
0.47930.92350.25450.96094
0.29310.95260.17200.97555
0.21090.96400.12460.97916

Framework versions

  • Transformers 4.51.3
  • TensorFlow 2.18.0
  • Datasets 3.6.0
  • Tokenizers 0.21.1