smt8731 commited on
Commit
19d6527
1 Parent(s): 0e7b54a

Model save

Browse files
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Rostlab/prot_bert
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - accuracy
7
+ - f1
8
+ model-index:
9
+ - name: prot_bert-finetuned-15ALL_TR
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # prot_bert-finetuned-15ALL_TR
17
+
18
+ This model is a fine-tuned version of [Rostlab/prot_bert](https://huggingface.co/Rostlab/prot_bert) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.6087
21
+ - Accuracy: 0.6696
22
+ - F1: 0.6689
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 2e-05
42
+ - train_batch_size: 64
43
+ - eval_batch_size: 64
44
+ - seed: 42
45
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
+ - lr_scheduler_type: linear
47
+ - num_epochs: 2
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
52
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
53
+ | 0.6715 | 1.0 | 186 | 0.6318 | 0.6528 | 0.6507 |
54
+ | 0.5998 | 2.0 | 372 | 0.6087 | 0.6696 | 0.6689 |
55
+
56
+
57
+ ### Framework versions
58
+
59
+ - Transformers 4.39.3
60
+ - Pytorch 2.2.1+cu121
61
+ - Datasets 2.18.0
62
+ - Tokenizers 0.15.2
config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "Rostlab/prot_bert",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout": null,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 1024,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 4096,
13
+ "layer_norm_eps": 1e-12,
14
+ "max_position_embeddings": 40000,
15
+ "model_type": "bert",
16
+ "num_attention_heads": 16,
17
+ "num_hidden_layers": 30,
18
+ "pad_token_id": 0,
19
+ "position_embedding_type": "absolute",
20
+ "problem_type": "single_label_classification",
21
+ "torch_dtype": "float32",
22
+ "transformers_version": "4.39.3",
23
+ "type_vocab_size": 2,
24
+ "use_cache": true,
25
+ "vocab_size": 30
26
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68d7c1f3cc6e41bcc113fe81e71afa2148dceb6cb076d3daaf36ad38e18aeb8f
3
+ size 1679791488
runs/Apr07_15-20-22_b6159c3c70de/events.out.tfevents.1712503261.b6159c3c70de.1492.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cada455c92b44f1543090ab68902e169b0f2a9ecd5691805efcec70fcbf2ca8b
3
+ size 6124
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
@@ -0,0 +1,187 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": "1.0",
3
+ "truncation": null,
4
+ "padding": {
5
+ "strategy": "BatchLongest",
6
+ "direction": "Right",
7
+ "pad_to_multiple_of": null,
8
+ "pad_id": 0,
9
+ "pad_type_id": 0,
10
+ "pad_token": "[PAD]"
11
+ },
12
+ "added_tokens": [
13
+ {
14
+ "id": 0,
15
+ "content": "[PAD]",
16
+ "single_word": false,
17
+ "lstrip": false,
18
+ "rstrip": false,
19
+ "normalized": false,
20
+ "special": true
21
+ },
22
+ {
23
+ "id": 1,
24
+ "content": "[UNK]",
25
+ "single_word": false,
26
+ "lstrip": false,
27
+ "rstrip": false,
28
+ "normalized": false,
29
+ "special": true
30
+ },
31
+ {
32
+ "id": 2,
33
+ "content": "[CLS]",
34
+ "single_word": false,
35
+ "lstrip": false,
36
+ "rstrip": false,
37
+ "normalized": false,
38
+ "special": true
39
+ },
40
+ {
41
+ "id": 3,
42
+ "content": "[SEP]",
43
+ "single_word": false,
44
+ "lstrip": false,
45
+ "rstrip": false,
46
+ "normalized": false,
47
+ "special": true
48
+ },
49
+ {
50
+ "id": 4,
51
+ "content": "[MASK]",
52
+ "single_word": false,
53
+ "lstrip": false,
54
+ "rstrip": false,
55
+ "normalized": false,
56
+ "special": true
57
+ }
58
+ ],
59
+ "normalizer": {
60
+ "type": "BertNormalizer",
61
+ "clean_text": true,
62
+ "handle_chinese_chars": true,
63
+ "strip_accents": null,
64
+ "lowercase": false
65
+ },
66
+ "pre_tokenizer": {
67
+ "type": "BertPreTokenizer"
68
+ },
69
+ "post_processor": {
70
+ "type": "TemplateProcessing",
71
+ "single": [
72
+ {
73
+ "SpecialToken": {
74
+ "id": "[CLS]",
75
+ "type_id": 0
76
+ }
77
+ },
78
+ {
79
+ "Sequence": {
80
+ "id": "A",
81
+ "type_id": 0
82
+ }
83
+ },
84
+ {
85
+ "SpecialToken": {
86
+ "id": "[SEP]",
87
+ "type_id": 0
88
+ }
89
+ }
90
+ ],
91
+ "pair": [
92
+ {
93
+ "SpecialToken": {
94
+ "id": "[CLS]",
95
+ "type_id": 0
96
+ }
97
+ },
98
+ {
99
+ "Sequence": {
100
+ "id": "A",
101
+ "type_id": 0
102
+ }
103
+ },
104
+ {
105
+ "SpecialToken": {
106
+ "id": "[SEP]",
107
+ "type_id": 0
108
+ }
109
+ },
110
+ {
111
+ "Sequence": {
112
+ "id": "B",
113
+ "type_id": 1
114
+ }
115
+ },
116
+ {
117
+ "SpecialToken": {
118
+ "id": "[SEP]",
119
+ "type_id": 1
120
+ }
121
+ }
122
+ ],
123
+ "special_tokens": {
124
+ "[CLS]": {
125
+ "id": "[CLS]",
126
+ "ids": [
127
+ 2
128
+ ],
129
+ "tokens": [
130
+ "[CLS]"
131
+ ]
132
+ },
133
+ "[SEP]": {
134
+ "id": "[SEP]",
135
+ "ids": [
136
+ 3
137
+ ],
138
+ "tokens": [
139
+ "[SEP]"
140
+ ]
141
+ }
142
+ }
143
+ },
144
+ "decoder": {
145
+ "type": "WordPiece",
146
+ "prefix": "##",
147
+ "cleanup": true
148
+ },
149
+ "model": {
150
+ "type": "WordPiece",
151
+ "unk_token": "[UNK]",
152
+ "continuing_subword_prefix": "##",
153
+ "max_input_chars_per_word": 100,
154
+ "vocab": {
155
+ "[PAD]": 0,
156
+ "[UNK]": 1,
157
+ "[CLS]": 2,
158
+ "[SEP]": 3,
159
+ "[MASK]": 4,
160
+ "L": 5,
161
+ "A": 6,
162
+ "G": 7,
163
+ "V": 8,
164
+ "E": 9,
165
+ "S": 10,
166
+ "I": 11,
167
+ "K": 12,
168
+ "R": 13,
169
+ "D": 14,
170
+ "T": 15,
171
+ "P": 16,
172
+ "N": 17,
173
+ "Q": 18,
174
+ "F": 19,
175
+ "Y": 20,
176
+ "M": 21,
177
+ "H": 22,
178
+ "C": 23,
179
+ "W": 24,
180
+ "X": 25,
181
+ "U": 26,
182
+ "B": 27,
183
+ "Z": 28,
184
+ "O": 29
185
+ }
186
+ }
187
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": false,
48
+ "full_tokenizer_file": null,
49
+ "mask_token": "[MASK]",
50
+ "model_max_length": 1000000000000000019884624838656,
51
+ "never_split": null,
52
+ "pad_token": "[PAD]",
53
+ "sep_token": "[SEP]",
54
+ "strip_accents": null,
55
+ "tokenize_chinese_chars": true,
56
+ "tokenizer_class": "BertTokenizer",
57
+ "unk_token": "[UNK]"
58
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:380d66f0c24df4830a1efbb4a036a6284986b2cef69e1a0d03925629ae93029e
3
+ size 4984
vocab.txt ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [PAD]
2
+ [UNK]
3
+ [CLS]
4
+ [SEP]
5
+ [MASK]
6
+ L
7
+ A
8
+ G
9
+ V
10
+ E
11
+ S
12
+ I
13
+ K
14
+ R
15
+ D
16
+ T
17
+ P
18
+ N
19
+ Q
20
+ F
21
+ Y
22
+ M
23
+ H
24
+ C
25
+ W
26
+ X
27
+ U
28
+ B
29
+ Z
30
+ O