CoolFace
Modelpublic

pyteach237/multilabel_lora_distilbert_classifier_tuned_ru

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

multilabelloradistilbertclassifiertuned_ru

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

  • Loss: 1.3658
  • Accuracy: 0.7845
  • F1: 0.7857
  • Precision: 0.7997
  • Recall: 0.7845

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: 4.993596574084884e-05
  • trainbatchsize: 4
  • evalbatchsize: 4
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • num_epochs: 15

Training results

Training LossEpochStepValidation LossAccuracyF1PrecisionRecall
1.06221.07270.90900.60250.59230.61490.6025
0.94492.014540.74510.68910.68550.69500.6891
0.70183.021810.61760.73590.73540.73770.7359
0.61924.029080.58540.77580.77510.78050.7758
0.49215.036350.57270.80610.80500.82020.8061
0.40916.043620.50190.82940.82930.83010.8294
0.32737.050890.48640.84040.84030.84090.8404
0.34738.058160.48280.85140.85120.85570.8514
0.28219.065430.46790.85970.85970.85970.8597
0.259910.072700.48740.88030.87990.88230.8803
0.271711.079970.45510.88310.88290.88320.8831
0.221112.087240.46020.88580.88560.88590.8858
0.220713.094510.50860.88450.88370.88620.8845
0.216614.0101780.47950.89410.89360.89520.8941
0.178215.0109050.46500.89550.89510.89590.8955

Framework versions

  • PEFT 0.11.1
  • Transformers 4.41.2
  • Pytorch 2.1.2
  • Datasets 2.19.2
  • Tokenizers 0.19.1