scenario-kd-po-ner-full-xlmr_data-univner_half55

This model is a fine-tuned version of haryoaw/scenario-TCR-NER_data-univner_half on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 53.5123
  • Precision: 0.7897
  • Recall: 0.7886
  • F1: 0.7891
  • Accuracy: 0.9790

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: 3e-05
  • train_batch_size: 8
  • eval_batch_size: 32
  • seed: 55
  • gradient_accumulation_steps: 4
  • total_train_batch_size: 32
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 10

Training results

Training Loss Epoch Step Validation Loss Precision Recall F1 Accuracy
94.1311 0.5828 500 77.4587 0.7447 0.7531 0.7489 0.9756
68.8855 1.1655 1000 69.8196 0.7632 0.7709 0.7670 0.9771
62.6609 1.7483 1500 66.1571 0.7615 0.7849 0.7730 0.9775
58.5525 2.3310 2000 63.3983 0.7699 0.7894 0.7795 0.9778
55.6843 2.9138 2500 61.2157 0.7885 0.7801 0.7843 0.9783
53.592 3.4965 3000 59.5858 0.7916 0.7769 0.7842 0.9786
51.7949 4.0793 3500 58.4899 0.7849 0.7823 0.7836 0.9785
50.4131 4.6620 4000 57.2748 0.7944 0.7868 0.7906 0.9787
49.2484 5.2448 4500 56.3953 0.7932 0.7823 0.7877 0.9788
48.3224 5.8275 5000 55.7812 0.7934 0.7774 0.7853 0.9783
47.5164 6.4103 5500 55.1109 0.7945 0.7826 0.7885 0.9788
46.895 6.9930 6000 54.5786 0.7885 0.7948 0.7916 0.9789
46.3036 7.5758 6500 54.3117 0.7943 0.7908 0.7926 0.9795
45.9459 8.1585 7000 53.9662 0.7912 0.7931 0.7921 0.9787
45.6374 8.7413 7500 53.7139 0.7917 0.7810 0.7863 0.9787
45.3582 9.3240 8000 53.6069 0.7904 0.7924 0.7914 0.9791
45.2705 9.9068 8500 53.5123 0.7897 0.7886 0.7891 0.9790

Framework versions

  • Transformers 4.44.2
  • Pytorch 2.1.1+cu121
  • Datasets 2.14.5
  • Tokenizers 0.19.1
Downloads last month
4
Safetensors
Model size
235M params
Tensor type
F32
·
Inference API
Unable to determine this model’s pipeline type. Check the docs .

Model tree for haryoaw/scenario-kd-po-ner-full-xlmr_data-univner_half55