eddiegulay commited on
Commit
9e83f05
1 Parent(s): 9a1b3d6

End of training

Browse files
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: distilbert/distilbert-base-uncased
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - conll2003
9
+ metrics:
10
+ - precision
11
+ - recall
12
+ - f1
13
+ - accuracy
14
+ model-index:
15
+ - name: base-NER
16
+ results:
17
+ - task:
18
+ name: Token Classification
19
+ type: token-classification
20
+ dataset:
21
+ name: conll2003
22
+ type: conll2003
23
+ config: conll2003
24
+ split: test
25
+ args: conll2003
26
+ metrics:
27
+ - name: Precision
28
+ type: precision
29
+ value: 0.8845085098992705
30
+ - name: Recall
31
+ type: recall
32
+ value: 0.9017351274787535
33
+ - name: F1
34
+ type: f1
35
+ value: 0.8930387515342801
36
+ - name: Accuracy
37
+ type: accuracy
38
+ value: 0.9782491655001615
39
+ ---
40
+
41
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
42
+ should probably proofread and complete it, then remove this comment. -->
43
+
44
+ # base-NER
45
+
46
+ This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the conll2003 dataset.
47
+ It achieves the following results on the evaluation set:
48
+ - Loss: 0.1129
49
+ - Precision: 0.8845
50
+ - Recall: 0.9017
51
+ - F1: 0.8930
52
+ - Accuracy: 0.9782
53
+
54
+ ## Model description
55
+
56
+ More information needed
57
+
58
+ ## Intended uses & limitations
59
+
60
+ More information needed
61
+
62
+ ## Training and evaluation data
63
+
64
+ More information needed
65
+
66
+ ## Training procedure
67
+
68
+ ### Training hyperparameters
69
+
70
+ The following hyperparameters were used during training:
71
+ - learning_rate: 2e-05
72
+ - train_batch_size: 16
73
+ - eval_batch_size: 16
74
+ - seed: 42
75
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
76
+ - lr_scheduler_type: linear
77
+ - num_epochs: 2
78
+
79
+ ### Training results
80
+
81
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
82
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
83
+ | 0.0595 | 1.0 | 878 | 0.1046 | 0.8676 | 0.8909 | 0.8791 | 0.9762 |
84
+ | 0.0319 | 2.0 | 1756 | 0.1129 | 0.8845 | 0.9017 | 0.8930 | 0.9782 |
85
+
86
+
87
+ ### Framework versions
88
+
89
+ - Transformers 4.44.2
90
+ - Pytorch 2.4.0+cu121
91
+ - Datasets 2.21.0
92
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c237f43a4d4b63e644fae23444b1833370a8b7a5c3a40cf2124e4ad8e04637d9
3
  size 265491548
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6e4e7f4699812fd80d38fb41f65b423f77228f20f7fb31020b1a322e9f12992
3
  size 265491548
runs/Sep10_20-35-10_594159c714ab/events.out.tfevents.1726000831.594159c714ab.264.2 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4c942e0418462031b37c7d40293855b5556dfa3d0fbf079c678f6b9ab7ac15af
3
- size 6221
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc0d20ab53db96d0aff2d5642ad03ccfb378c8f8bdc038b8f06fc11e934723c9
3
+ size 7047