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-distemist-dev-ner
+- multi-train-distemist-dev-ner
+metrics:
+- precision
+- recall
+- f1
+- accuracy
 model-index:
 - name: output
-  results: []
+  results:
+  - task:
+      name: Token Classification
+      type: token-classification
+    dataset:
+      name: multi-train-distemist-dev-ner
+      type: multi-train-distemist-dev-ner
+      config: MultiTrainDisTEMISTDevNER
+      split: validation
+      args: MultiTrainDisTEMISTDevNER
+    metrics:
+    - name: Precision
+      type: precision
+      value: 0.3145413870246085
+    - name: Recall
+      type: recall
+      value: 0.8224145999064109
+    - name: F1
+      type: f1
+      value: 0.45504563402161957
+    - name: Accuracy
+      type: accuracy
+      value: 0.8526409018818958
 ---
 
 <!-- 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-distemist-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-distemist-dev-ner dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 2.3735
-- eval_precision: 0.0045
-- eval_recall: 0.1273
-- eval_f1: 0.0088
-- eval_accuracy: 0.0187
-- eval_runtime: 16.9354
-- eval_samples_per_second: 401.939
-- eval_steps_per_second: 50.25
-- step: 0
+- Loss: 1.1188
+- Precision: 0.3145
+- Recall: 0.8224
+- F1: 0.4550
+- Accuracy: 0.8526
 
 ## 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.4319          | 0.2617    | 0.7866 | 0.3927 | 0.8359   |
+| 0.1853        | 2.0    | 3403  | 0.3841          | 0.3142    | 0.7829 | 0.4485 | 0.8645   |
+| 0.1254        | 2.9997 | 5104  | 0.6410          | 0.3055    | 0.8088 | 0.4435 | 0.8436   |
+| 0.0823        | 4.0    | 6806  | 0.7242          | 0.2964    | 0.8074 | 0.4336 | 0.8436   |
+| 0.0597        | 4.9997 | 8507  | 0.7756          | 0.3133    | 0.7948 | 0.4495 | 0.8502   |
+| 0.0446        | 6.0    | 10209 | 0.8561          | 0.3137    | 0.8037 | 0.4513 | 0.8483   |
+| 0.0325        | 6.9997 | 11910 | 0.9499          | 0.3214    | 0.8278 | 0.4631 | 0.8560   |
+| 0.022         | 8.0    | 13612 | 1.0452          | 0.3129    | 0.8222 | 0.4533 | 0.8510   |
+| 0.017         | 8.9997 | 15313 | 1.1025          | 0.3133    | 0.8180 | 0.4531 | 0.8524   |
+| 0.0135        | 9.9971 | 17010 | 1.1188          | 0.3145    | 0.8224 | 0.4550 | 0.8526   |
+
+
 ### 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:b365380c816f8752742eecc6b07b7e1bbabadcc327738bcebf63b984484dcd2e
 size 496262556

train.log

@@ -1630,3 +1630,13 @@ Training completed. Do not forget to share your model on huggingface.co/models =
 [INFO|trainer.py:2521] 2024-05-13 12:31:23,201 >> Loading best model from /content/dissertation/scripts/ner/output/checkpoint-11910 (score: 0.46305870034683594).
 
                                                     
 [INFO|trainer.py:4035] 2024-05-13 12:31:23,414 >> Waiting for the current checkpoint push to be finished, this might take a couple of minutes.
+[INFO|trainer.py:3305] 2024-05-13 12:32:21,076 >> Saving model checkpoint to /content/dissertation/scripts/ner/output
+[INFO|configuration_utils.py:471] 2024-05-13 12:32:21,078 >> Configuration saved in /content/dissertation/scripts/ner/output/config.json
+[INFO|modeling_utils.py:2590] 2024-05-13 12:32:22,291 >> Model weights saved in /content/dissertation/scripts/ner/output/model.safetensors
+[INFO|tokenization_utils_base.py:2488] 2024-05-13 12:32:22,292 >> tokenizer config file saved in /content/dissertation/scripts/ner/output/tokenizer_config.json
+[INFO|tokenization_utils_base.py:2497] 2024-05-13 12:32:22,293 >> Special tokens file saved in /content/dissertation/scripts/ner/output/special_tokens_map.json
+[INFO|trainer.py:3305] 2024-05-13 12:32:22,344 >> Saving model checkpoint to /content/dissertation/scripts/ner/output
+[INFO|configuration_utils.py:471] 2024-05-13 12:32:22,345 >> Configuration saved in /content/dissertation/scripts/ner/output/config.json
+[INFO|modeling_utils.py:2590] 2024-05-13 12:32:23,238 >> Model weights saved in /content/dissertation/scripts/ner/output/model.safetensors
+[INFO|tokenization_utils_base.py:2488] 2024-05-13 12:32:23,239 >> tokenizer config file saved in /content/dissertation/scripts/ner/output/tokenizer_config.json
+[INFO|tokenization_utils_base.py:2497] 2024-05-13 12:32:23,239 >> Special tokens file saved in /content/dissertation/scripts/ner/output/special_tokens_map.json