Training complete

README.md

@@ -1,6 +1,13 @@
 ---
+license: apache-2.0
+base_model: distilbert/distilbert-base-cased
 tags:
 - generated_from_trainer
+metrics:
+- precision
+- recall
+- f1
+- accuracy
 model-index:
 - name: distillbert-finetuned-ner-btc
   results: []
@@ -11,7 +18,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distillbert-finetuned-ner-btc
 
-This model was trained from scratch on an unknown dataset.
+This model is a fine-tuned version of [distilbert/distilbert-base-cased](https://huggingface.co/distilbert/distilbert-base-cased) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2802
+- Precision: 0.4392
+- Recall: 0.4370
+- F1: 0.4381
+- Accuracy: 0.9117
 
 ## Model description
 
@@ -36,13 +49,15 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 1
+- num_epochs: 3
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1  | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:---:|:--------:|
-| No log        | 1.0   | 56   | 0.4471          | 0.0       | 0.0    | 0.0 | 0.8574   |
+| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
+| No log        | 1.0   | 56   | 0.3448          | 0.4762    | 0.0345 | 0.0644 | 0.8792   |
+| No log        | 2.0   | 112  | 0.2880          | 0.4506    | 0.3938 | 0.4203 | 0.9080   |
+| No log        | 3.0   | 168  | 0.2802          | 0.4392    | 0.4370 | 0.4381 | 0.9117   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:63de48e0b0c12e02e07823a3c0a47408948d9ab6cb12c797c65a6032791a6f33
+oid sha256:32b832876674cc5c4d00a1b29060d5cba3fbd2975758dc72e3e08fc0b68d4836
 size 260797516