balciberin commited on
Commit
9018396
1 Parent(s): d080390

End of training

Browse files
Files changed (2) hide show
  1. README.md +99 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: distilbert-base-uncased
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - wnut_17
8
+ metrics:
9
+ - precision
10
+ - recall
11
+ - f1
12
+ - accuracy
13
+ model-index:
14
+ - name: ner_model
15
+ results:
16
+ - task:
17
+ name: Token Classification
18
+ type: token-classification
19
+ dataset:
20
+ name: wnut_17
21
+ type: wnut_17
22
+ config: wnut_17
23
+ split: test
24
+ args: wnut_17
25
+ metrics:
26
+ - name: Precision
27
+ type: precision
28
+ value: 0.5632040050062578
29
+ - name: Recall
30
+ type: recall
31
+ value: 0.4170528266913809
32
+ - name: F1
33
+ type: f1
34
+ value: 0.47923322683706066
35
+ - name: Accuracy
36
+ type: accuracy
37
+ value: 0.9478859390363815
38
+ ---
39
+
40
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
41
+ should probably proofread and complete it, then remove this comment. -->
42
+
43
+ # ner_model
44
+
45
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the wnut_17 dataset.
46
+ It achieves the following results on the evaluation set:
47
+ - Loss: 0.3832
48
+ - Precision: 0.5632
49
+ - Recall: 0.4171
50
+ - F1: 0.4792
51
+ - Accuracy: 0.9479
52
+
53
+ ## Model description
54
+
55
+ More information needed
56
+
57
+ ## Intended uses & limitations
58
+
59
+ More information needed
60
+
61
+ ## Training and evaluation data
62
+
63
+ More information needed
64
+
65
+ ## Training procedure
66
+
67
+ ### Training hyperparameters
68
+
69
+ The following hyperparameters were used during training:
70
+ - learning_rate: 2e-05
71
+ - train_batch_size: 8
72
+ - eval_batch_size: 8
73
+ - seed: 42
74
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
75
+ - lr_scheduler_type: linear
76
+ - num_epochs: 10
77
+
78
+ ### Training results
79
+
80
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
81
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
82
+ | No log | 1.0 | 425 | 0.2828 | 0.6021 | 0.3800 | 0.4659 | 0.9466 |
83
+ | 0.074 | 2.0 | 850 | 0.2955 | 0.5825 | 0.3892 | 0.4667 | 0.9474 |
84
+ | 0.0457 | 3.0 | 1275 | 0.3072 | 0.5857 | 0.4180 | 0.4878 | 0.9492 |
85
+ | 0.0234 | 4.0 | 1700 | 0.3430 | 0.5911 | 0.4059 | 0.4813 | 0.9481 |
86
+ | 0.0144 | 5.0 | 2125 | 0.3468 | 0.5406 | 0.4198 | 0.4726 | 0.9476 |
87
+ | 0.0107 | 6.0 | 2550 | 0.3742 | 0.5541 | 0.4032 | 0.4667 | 0.9470 |
88
+ | 0.0107 | 7.0 | 2975 | 0.3779 | 0.5861 | 0.4133 | 0.4848 | 0.9483 |
89
+ | 0.0081 | 8.0 | 3400 | 0.3802 | 0.5537 | 0.4013 | 0.4653 | 0.9477 |
90
+ | 0.0059 | 9.0 | 3825 | 0.3750 | 0.5511 | 0.4198 | 0.4766 | 0.9478 |
91
+ | 0.0033 | 10.0 | 4250 | 0.3832 | 0.5632 | 0.4171 | 0.4792 | 0.9479 |
92
+
93
+
94
+ ### Framework versions
95
+
96
+ - Transformers 4.41.2
97
+ - Pytorch 2.3.0+cu121
98
+ - Datasets 2.19.2
99
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3dcb48549d2fadd03f116654f7a89f0e0bab7502fda3e9a25e33b7a0cb151ec5
3
  size 265503852
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e65549fd9f90d1a5e9ae1b52c4a82b51bd9de9f276459be4474fe065b553240
3
  size 265503852