xlm-base-CONTINUE-finetuned-ner-covidmed-v1
This model is a fine-tuned version of quanxuantruong/xlm-base-finetuned-ner-covidmed-v2 on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 0.0794
- Accuracy: 0.9802
- Precision: 0.8812
- Recall: 0.9253
- F1: 0.9015
- Age Precision: 0.9509
- Age Recall: 0.9656
- Age F1-score: 0.9582
- Date Precision: 0.9820
- Date Recall: 0.9915
- Date F1-score: 0.9868
- Gender Precision: 0.9350
- Gender Recall: 0.9654
- Gender F1-score: 0.9499
- Job Precision: 0.5210
- Job Recall: 0.7168
- Job F1-score: 0.6034
- Location Precision: 0.9356
- Location Recall: 0.9520
- Location F1-score: 0.9438
- Name Precision: 0.8988
- Name Recall: 0.9214
- Name F1-score: 0.9099
- Organization Precision: 0.8475
- Organization Recall: 0.9079
- Organization F1-score: 0.8766
- Patient Id Precision: 0.9686
- Patient Id Recall: 0.9840
- Patient Id F1-score: 0.9762
- Symptom And Disease Precision: 0.8482
- Symptom And Disease Recall: 0.8952
- Symptom And Disease F1-score: 0.8711
- Transportation Precision: 0.9246
- Transportation Recall: 0.9534
- Transportation F1-score: 0.9388
- Micro avg Precision: 0.9243
- Micro avg Recall: 0.9516
- Micro avg F1-score: 0.9377
- Macro avg Precision: 0.8812
- Macro avg Recall: 0.9253
- Macro avg F1-score: 0.9015
- Weighted avg Precision: 0.9270
- Weighted avg Recall: 0.9516
- Weighted avg F1-score: 0.9389
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
- train_batch_size: 64
- eval_batch_size: 64
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 5
Training results
Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 | Age Precision | Age Recall | Age F1-score | Date Precision | Date Recall | Date F1-score | Gender Precision | Gender Recall | Gender F1-score | Job Precision | Job Recall | Job F1-score | Location Precision | Location Recall | Location F1-score | Name Precision | Name Recall | Name F1-score | Organization Precision | Organization Recall | Organization F1-score | Patient Id Precision | Patient Id Recall | Patient Id F1-score | Symptom And Disease Precision | Symptom And Disease Recall | Symptom And Disease F1-score | Transportation Precision | Transportation Recall | Transportation F1-score | Micro avg Precision | Micro avg Recall | Micro avg F1-score | Macro avg Precision | Macro avg Recall | Macro avg F1-score | Weighted avg Precision | Weighted avg Recall | Weighted avg F1-score |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
No log | 1.0 | 32 | 0.0859 | 0.9755 | 0.7912 | 0.8950 | 0.8378 | 0.8710 | 0.9863 | 0.9251 | 0.9733 | 0.9915 | 0.9823 | 0.8407 | 0.9827 | 0.9062 | 0.4466 | 0.6763 | 0.5379 | 0.9129 | 0.9412 | 0.9268 | 0.6943 | 0.8428 | 0.7614 | 0.7932 | 0.9105 | 0.8478 | 0.9644 | 0.9850 | 0.9746 | 0.816 | 0.8979 | 0.8550 | 0.5992 | 0.7358 | 0.6605 | 0.8836 | 0.9435 | 0.9126 | 0.7912 | 0.8950 | 0.8378 | 0.8901 | 0.9435 | 0.9152 |
No log | 2.0 | 64 | 0.0818 | 0.9771 | 0.8266 | 0.9005 | 0.8599 | 0.9337 | 0.9674 | 0.9502 | 0.9814 | 0.9915 | 0.9865 | 0.9119 | 0.9632 | 0.9368 | 0.5020 | 0.7225 | 0.5924 | 0.9347 | 0.9482 | 0.9414 | 0.8182 | 0.8774 | 0.8467 | 0.8251 | 0.9053 | 0.8633 | 0.9709 | 0.9830 | 0.9770 | 0.8186 | 0.9058 | 0.8600 | 0.5697 | 0.7409 | 0.6441 | 0.9071 | 0.9462 | 0.9263 | 0.8266 | 0.9005 | 0.8599 | 0.9126 | 0.9462 | 0.9285 |
No log | 3.0 | 96 | 0.0798 | 0.9795 | 0.8542 | 0.9065 | 0.8781 | 0.945 | 0.9742 | 0.9594 | 0.9820 | 0.9921 | 0.9871 | 0.9366 | 0.9589 | 0.9476 | 0.5021 | 0.6936 | 0.5825 | 0.9316 | 0.9536 | 0.9425 | 0.8916 | 0.9057 | 0.8986 | 0.8548 | 0.8859 | 0.8701 | 0.9672 | 0.9840 | 0.9755 | 0.8416 | 0.8979 | 0.8688 | 0.6900 | 0.8187 | 0.7488 | 0.9172 | 0.9483 | 0.9325 | 0.8542 | 0.9065 | 0.8781 | 0.9205 | 0.9483 | 0.9339 |
No log | 4.0 | 128 | 0.0790 | 0.9800 | 0.8821 | 0.9252 | 0.9018 | 0.9557 | 0.9639 | 0.9598 | 0.9826 | 0.9915 | 0.9871 | 0.9387 | 0.9610 | 0.9497 | 0.5274 | 0.7225 | 0.6098 | 0.9339 | 0.9514 | 0.9426 | 0.8988 | 0.9214 | 0.9099 | 0.8464 | 0.9079 | 0.8761 | 0.9695 | 0.9840 | 0.9767 | 0.8432 | 0.8996 | 0.8705 | 0.9242 | 0.9482 | 0.9361 | 0.9237 | 0.9515 | 0.9374 | 0.8821 | 0.9252 | 0.9018 | 0.9265 | 0.9515 | 0.9386 |
No log | 5.0 | 160 | 0.0794 | 0.9802 | 0.8812 | 0.9253 | 0.9015 | 0.9509 | 0.9656 | 0.9582 | 0.9820 | 0.9915 | 0.9868 | 0.9350 | 0.9654 | 0.9499 | 0.5210 | 0.7168 | 0.6034 | 0.9356 | 0.9520 | 0.9438 | 0.8988 | 0.9214 | 0.9099 | 0.8475 | 0.9079 | 0.8766 | 0.9686 | 0.9840 | 0.9762 | 0.8482 | 0.8952 | 0.8711 | 0.9246 | 0.9534 | 0.9388 | 0.9243 | 0.9516 | 0.9377 | 0.8812 | 0.9253 | 0.9015 | 0.9270 | 0.9516 | 0.9389 |
Framework versions
- Transformers 4.51.3
- Pytorch 2.6.0+cu124
- Datasets 3.6.0
- Tokenizers 0.21.1
- Downloads last month
- 2
Inference Providers
NEW
This model isn't deployed by any Inference Provider.
🙋
Ask for provider support
Model tree for quanxuantruong/xlm-base-CONTINUE-finetuned-ner-covidmed-v1
Base model
FacebookAI/xlm-roberta-base