Hugging Face's logo

cehongw
/
ner-fine-tune-roberta-new

Token Classification
Transformers
PyTorch
roberta
Generated from Trainer
Inference Endpoints
Model card Files Community
cehongw commited on
Commit
fa4d265
•
1 Parent(s): 422b504

Training complete

Browse files
Files changed (2) hide show
  1. README.md +95 -0
  2. pytorch_model.bin +1 -1
README.md ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ base_model: roberta-base
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - precision
8
+ - recall
9
+ - f1
10
+ - accuracy
11
+ model-index:
12
+ - name: ner-fine-tune-roberta-new
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
+ # ner-fine-tune-roberta-new
20
+
21
+ This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.3320
24
+ - Precision: 0.2696
25
+ - Recall: 0.3767
26
+ - F1: 0.3143
27
+ - Accuracy: 0.9389
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: 1e-05
47
+ - train_batch_size: 8
48
+ - eval_batch_size: 8
49
+ - seed: 42
50
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
+ - lr_scheduler_type: linear
52
+ - num_epochs: 30
53
+
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
+ | No log | 1.0 | 122 | 0.2264 | 0.0 | 0.0 | 0.0 | 0.9542 |
59
+ | No log | 2.0 | 244 | 0.1874 | 0.2348 | 0.1349 | 0.1713 | 0.9571 |
60
+ | No log | 3.0 | 366 | 0.1739 | 0.2420 | 0.2279 | 0.2347 | 0.9528 |
61
+ | No log | 4.0 | 488 | 0.1656 | 0.1939 | 0.2233 | 0.2076 | 0.9472 |
62
+ | 0.2548 | 5.0 | 610 | 0.1740 | 0.3243 | 0.2767 | 0.2986 | 0.9573 |
63
+ | 0.2548 | 6.0 | 732 | 0.2087 | 0.2562 | 0.2628 | 0.2595 | 0.9464 |
64
+ | 0.2548 | 7.0 | 854 | 0.1921 | 0.2773 | 0.2953 | 0.2860 | 0.9495 |
65
+ | 0.2548 | 8.0 | 976 | 0.2038 | 0.2602 | 0.3860 | 0.3109 | 0.9397 |
66
+ | 0.0748 | 9.0 | 1098 | 0.2324 | 0.2371 | 0.3093 | 0.2684 | 0.9398 |
67
+ | 0.0748 | 10.0 | 1220 | 0.2329 | 0.2852 | 0.3442 | 0.3119 | 0.9436 |
68
+ | 0.0748 | 11.0 | 1342 | 0.2670 | 0.2521 | 0.3535 | 0.2943 | 0.9356 |
69
+ | 0.0748 | 12.0 | 1464 | 0.2607 | 0.2509 | 0.3186 | 0.2807 | 0.9395 |
70
+ | 0.033 | 13.0 | 1586 | 0.2645 | 0.2655 | 0.3791 | 0.3123 | 0.9359 |
71
+ | 0.033 | 14.0 | 1708 | 0.2947 | 0.2838 | 0.4442 | 0.3463 | 0.9398 |
72
+ | 0.033 | 15.0 | 1830 | 0.2807 | 0.2945 | 0.3349 | 0.3134 | 0.9451 |
73
+ | 0.033 | 16.0 | 1952 | 0.2990 | 0.2910 | 0.3302 | 0.3094 | 0.9448 |
74
+ | 0.0181 | 17.0 | 2074 | 0.2915 | 0.2799 | 0.3651 | 0.3169 | 0.9425 |
75
+ | 0.0181 | 18.0 | 2196 | 0.2853 | 0.2868 | 0.3535 | 0.3167 | 0.9424 |
76
+ | 0.0181 | 19.0 | 2318 | 0.2991 | 0.2918 | 0.3814 | 0.3306 | 0.9440 |
77
+ | 0.0181 | 20.0 | 2440 | 0.2863 | 0.2762 | 0.3744 | 0.3179 | 0.9408 |
78
+ | 0.0111 | 21.0 | 2562 | 0.3280 | 0.2796 | 0.3628 | 0.3158 | 0.9409 |
79
+ | 0.0111 | 22.0 | 2684 | 0.3135 | 0.2772 | 0.3372 | 0.3043 | 0.9431 |
80
+ | 0.0111 | 23.0 | 2806 | 0.3282 | 0.2632 | 0.3698 | 0.3075 | 0.9404 |
81
+ | 0.0111 | 24.0 | 2928 | 0.3306 | 0.2597 | 0.3884 | 0.3113 | 0.9369 |
82
+ | 0.0078 | 25.0 | 3050 | 0.3135 | 0.2743 | 0.3605 | 0.3116 | 0.9414 |
83
+ | 0.0078 | 26.0 | 3172 | 0.3320 | 0.2646 | 0.3791 | 0.3117 | 0.9374 |
84
+ | 0.0078 | 27.0 | 3294 | 0.3273 | 0.2659 | 0.3791 | 0.3126 | 0.9381 |
85
+ | 0.0078 | 28.0 | 3416 | 0.3290 | 0.2616 | 0.3674 | 0.3056 | 0.9380 |
86
+ | 0.0055 | 29.0 | 3538 | 0.3366 | 0.2656 | 0.3860 | 0.3147 | 0.9372 |
87
+ | 0.0055 | 30.0 | 3660 | 0.3320 | 0.2696 | 0.3767 | 0.3143 | 0.9389 |
88
+
89
+
90
+ ### Framework versions
91
+
92
+ - Transformers 4.34.1
93
+ - Pytorch 2.1.0+cu118
94
+ - Datasets 2.14.6
95
+ - Tokenizers 0.14.1
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1e9c9d36efb8420bf4219b6db36602159e92b651d82dead720f14c0d228d995d
3
  size 496335078
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79a06049c409a51febda71e0ef0f14e917f254e5a9fdf288a9788eb097ca49af
3
  size 496335078