End of training

README.md

@@ -1,9 +1,14 @@
 ---
 library_name: transformers
 license: apache-2.0
-base_model: distilbert/distilbert-base-uncased
+base_model: dany0407/token_classification_NER
 tags:
 - generated_from_trainer
+metrics:
+- precision
+- recall
+- f1
+- accuracy
 model-index:
 - name: token_classification_NER
   results: []
@@ -14,7 +19,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # token_classification_NER
 
-This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [dany0407/token_classification_NER](https://huggingface.co/dany0407/token_classification_NER) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.3086
+- Precision: 0.5215
+- Recall: 0.3939
+- F1: 0.4488
+- Accuracy: 0.9461
 
 ## Model description
 
@@ -39,13 +50,17 @@ The following hyperparameters were used during training:
 - 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: 1
+- num_epochs: 5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| No log        | 1.0   | 213  | 0.3104          | 0.3900    | 0.0871 | 0.1424 | 0.9316   |
+| No log        | 1.0   | 213  | 0.2594          | 0.6151    | 0.3095 | 0.4118 | 0.9428   |
+| No log        | 2.0   | 426  | 0.2813          | 0.5541    | 0.3466 | 0.4265 | 0.9444   |
+| 0.0965        | 3.0   | 639  | 0.3013          | 0.5584    | 0.3587 | 0.4368 | 0.9455   |
+| 0.0965        | 4.0   | 852  | 0.3036          | 0.5416    | 0.3865 | 0.4511 | 0.9463   |
+| 0.0375        | 5.0   | 1065 | 0.3086          | 0.5215    | 0.3939 | 0.4488 | 0.9461   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c6795bd498416f75e05722dc9044e070e65082a2334f51d4e85a8e7a796ea99d
+oid sha256:bb5a3c4ace9ecb3ce7777c9e949482c6067b7d1368d480460f0ffcdc999e050c
 size 265503852