imrazaa commited on
Commit
bc0ce72
1 Parent(s): d815029

End of training

Browse files
Files changed (2) hide show
  1. README.md +68 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: distilbert-base-uncased
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - precision
8
+ - recall
9
+ - f1
10
+ - accuracy
11
+ model-index:
12
+ - name: named-entity-recognition-distilbert-B
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # named-entity-recognition-distilbert-B
20
+
21
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.0350
24
+ - Precision: 0.9377
25
+ - Recall: 0.9442
26
+ - F1: 0.9409
27
+ - Accuracy: 0.9918
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 2e-05
47
+ - train_batch_size: 32
48
+ - eval_batch_size: 16
49
+ - seed: 42
50
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
+ - lr_scheduler_type: linear
52
+ - num_epochs: 3
53
+
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
+ |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
+ | 0.0164 | 1.0 | 8205 | 0.0256 | 0.9347 | 0.9385 | 0.9366 | 0.9913 |
59
+ | 0.0082 | 2.0 | 16410 | 0.0288 | 0.9381 | 0.9413 | 0.9397 | 0.9917 |
60
+ | 0.0044 | 3.0 | 24615 | 0.0350 | 0.9377 | 0.9442 | 0.9409 | 0.9918 |
61
+
62
+
63
+ ### Framework versions
64
+
65
+ - Transformers 4.35.2
66
+ - Pytorch 2.1.1+cu121
67
+ - Datasets 2.15.0
68
+ - Tokenizers 0.15.0
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:514227011231b11da42e2fec78a98f631ab4dd489aca74b64934ab118795c08a
3
  size 265559228
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1eef5bed1ea78439cf739ca32156e49759ffc2ec31564617fe7f57cd54c97470
3
  size 265559228