Model save

README.md

@@ -2,13 +2,39 @@
 license: apache-2.0
 base_model: PlanTL-GOB-ES/bsc-bio-ehr-es
 tags:
-- token-classification
 - generated_from_trainer
 datasets:
-- Rodrigo1771/multi-train-drugtemist-dev-ner
+- multi-train-drugtemist-dev-ner
+metrics:
+- precision
+- recall
+- f1
+- accuracy
 model-index:
 - name: output
-  results: []
+  results:
+  - task:
+      name: Token Classification
+      type: token-classification
+    dataset:
+      name: multi-train-drugtemist-dev-ner
+      type: multi-train-drugtemist-dev-ner
+      config: MultiTrainDrugTEMISTDevNER
+      split: validation
+      args: MultiTrainDrugTEMISTDevNER
+    metrics:
+    - name: Precision
+      type: precision
+      value: 0.09270693512304251
+    - name: Recall
+      type: recall
+      value: 0.9522058823529411
+    - name: F1
+      type: f1
+      value: 0.16896354888689555
+    - name: Accuracy
+      type: accuracy
+      value: 0.7845534874460183
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,17 +42,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # output
 
-This model is a fine-tuned version of [PlanTL-GOB-ES/bsc-bio-ehr-es](https://huggingface.co/PlanTL-GOB-ES/bsc-bio-ehr-es) on the Rodrigo1771/multi-train-drugtemist-dev-ner dataset.
+This model is a fine-tuned version of [PlanTL-GOB-ES/bsc-bio-ehr-es](https://huggingface.co/PlanTL-GOB-ES/bsc-bio-ehr-es) on the multi-train-drugtemist-dev-ner dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 2.4031
-- eval_precision: 0.0004
-- eval_recall: 0.0386
-- eval_f1: 0.0007
-- eval_accuracy: 0.0028
-- eval_runtime: 16.7962
-- eval_samples_per_second: 405.27
-- eval_steps_per_second: 50.666
-- step: 0
+- Loss: 1.7861
+- Precision: 0.0927
+- Recall: 0.9522
+- F1: 0.1690
+- Accuracy: 0.7846
 
 ## Model description
 
@@ -55,6 +77,22 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 10.0
 
+### Training results
+
+| Training Loss | Epoch  | Step  | Validation Loss | Precision | Recall | F1     | Accuracy |
+|:-------------:|:------:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
+| 0.2596        | 0.9997 | 1701  | 0.7913          | 0.0793    | 0.9366 | 0.1462 | 0.7672   |
+| 0.1853        | 2.0    | 3403  | 0.6631          | 0.0969    | 0.9485 | 0.1759 | 0.8100   |
+| 0.1254        | 2.9997 | 5104  | 1.0729          | 0.0906    | 0.9421 | 0.1653 | 0.7755   |
+| 0.0823        | 4.0    | 6806  | 1.2568          | 0.0888    | 0.9504 | 0.1624 | 0.7719   |
+| 0.0597        | 4.9997 | 8507  | 1.1908          | 0.0941    | 0.9375 | 0.1710 | 0.7837   |
+| 0.0446        | 6.0    | 10209 | 1.3844          | 0.0944    | 0.9504 | 0.1718 | 0.7812   |
+| 0.0325        | 6.9997 | 11910 | 1.5515          | 0.0937    | 0.9476 | 0.1705 | 0.7866   |
+| 0.022         | 8.0    | 13612 | 1.6300          | 0.0926    | 0.9559 | 0.1689 | 0.7843   |
+| 0.017         | 8.9997 | 15313 | 1.7459          | 0.0929    | 0.9531 | 0.1693 | 0.7845   |
+| 0.0135        | 9.9971 | 17010 | 1.7861          | 0.0927    | 0.9522 | 0.1690 | 0.7846   |
+
+
 ### Framework versions
 
 - Transformers 4.40.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0931a4af01357abb6f7dc6ad8c0e58400dcf281e4ea89e471d6ca8c508ad5be3
+oid sha256:a92e9473c4569e5b9f9c2355692ca54a9ea2c391a97e47dae7d504f0230253df
 size 496262556

train.log

@@ -1603,3 +1603,13 @@ Training completed. Do not forget to share your model on huggingface.co/models =
 [INFO|trainer.py:2521] 2024-05-13 14:58:14,696 >> Loading best model from /content/dissertation/scripts/ner/output/checkpoint-3403 (score: 0.17586912065439672).
 
                                                     
 [INFO|trainer.py:4035] 2024-05-13 14:58:14,891 >> Waiting for the current checkpoint push to be finished, this might take a couple of minutes.
+[INFO|trainer.py:3305] 2024-05-13 14:58:37,779 >> Saving model checkpoint to /content/dissertation/scripts/ner/output
+[INFO|configuration_utils.py:471] 2024-05-13 14:58:37,781 >> Configuration saved in /content/dissertation/scripts/ner/output/config.json
+[INFO|modeling_utils.py:2590] 2024-05-13 14:58:39,085 >> Model weights saved in /content/dissertation/scripts/ner/output/model.safetensors
+[INFO|tokenization_utils_base.py:2488] 2024-05-13 14:58:39,087 >> tokenizer config file saved in /content/dissertation/scripts/ner/output/tokenizer_config.json
+[INFO|tokenization_utils_base.py:2497] 2024-05-13 14:58:39,087 >> Special tokens file saved in /content/dissertation/scripts/ner/output/special_tokens_map.json
+[INFO|trainer.py:3305] 2024-05-13 14:58:39,136 >> Saving model checkpoint to /content/dissertation/scripts/ner/output
+[INFO|configuration_utils.py:471] 2024-05-13 14:58:39,137 >> Configuration saved in /content/dissertation/scripts/ner/output/config.json
+[INFO|modeling_utils.py:2590] 2024-05-13 14:58:40,302 >> Model weights saved in /content/dissertation/scripts/ner/output/model.safetensors
+[INFO|tokenization_utils_base.py:2488] 2024-05-13 14:58:40,303 >> tokenizer config file saved in /content/dissertation/scripts/ner/output/tokenizer_config.json
+[INFO|tokenization_utils_base.py:2497] 2024-05-13 14:58:40,303 >> Special tokens file saved in /content/dissertation/scripts/ner/output/special_tokens_map.json