tanoManzo commited on
Commit
239b82a
·
verified ·
1 Parent(s): f4b2f83

End of training

Browse files
README.md ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: zhihan1996/DNABERT-2-117M
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - precision
7
+ - recall
8
+ - accuracy
9
+ model-index:
10
+ - name: DNABERT-2-117M_ft_BioS2_1kbpHG19_DHSs_H3K27AC
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # DNABERT-2-117M_ft_BioS2_1kbpHG19_DHSs_H3K27AC
18
+
19
+ This model is a fine-tuned version of [zhihan1996/DNABERT-2-117M](https://huggingface.co/zhihan1996/DNABERT-2-117M) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.4620
22
+ - F1 Score: 0.7974
23
+ - Precision: 0.8134
24
+ - Recall: 0.7819
25
+ - Accuracy: 0.7933
26
+ - Auc: 0.8741
27
+ - Prc: 0.8601
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: 16
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: 20
53
+ - mixed_precision_training: Native AMP
54
+
55
+ ### Training results
56
+
57
+ | Training Loss | Epoch | Step | Validation Loss | F1 Score | Precision | Recall | Accuracy | Auc | Prc |
58
+ |:-------------:|:------:|:----:|:---------------:|:--------:|:---------:|:------:|:--------:|:------:|:------:|
59
+ | 0.706 | 0.1680 | 500 | 0.6216 | 0.6107 | 0.7600 | 0.5105 | 0.6616 | 0.7690 | 0.7525 |
60
+ | 0.614 | 0.3360 | 1000 | 0.5784 | 0.7600 | 0.6685 | 0.8805 | 0.7108 | 0.7836 | 0.7638 |
61
+ | 0.5795 | 0.5040 | 1500 | 0.5409 | 0.7576 | 0.6964 | 0.8307 | 0.7236 | 0.7938 | 0.7723 |
62
+ | 0.557 | 0.6720 | 2000 | 0.5898 | 0.7622 | 0.6293 | 0.9661 | 0.6864 | 0.8063 | 0.7838 |
63
+ | 0.561 | 0.8401 | 2500 | 0.5143 | 0.7770 | 0.7298 | 0.8307 | 0.7520 | 0.8183 | 0.7953 |
64
+ | 0.5313 | 1.0081 | 3000 | 0.4928 | 0.7958 | 0.7300 | 0.8746 | 0.7666 | 0.8334 | 0.8096 |
65
+ | 0.5167 | 1.1761 | 3500 | 0.5086 | 0.7590 | 0.7963 | 0.7250 | 0.7605 | 0.8471 | 0.8266 |
66
+ | 0.5052 | 1.3441 | 4000 | 0.4732 | 0.8021 | 0.7578 | 0.8520 | 0.7814 | 0.8533 | 0.8327 |
67
+ | 0.4923 | 1.5121 | 4500 | 0.4741 | 0.7898 | 0.7864 | 0.7932 | 0.7804 | 0.8572 | 0.8386 |
68
+ | 0.493 | 1.6801 | 5000 | 0.4683 | 0.7821 | 0.7995 | 0.7654 | 0.7782 | 0.8618 | 0.8431 |
69
+ | 0.4834 | 1.8481 | 5500 | 0.4522 | 0.8100 | 0.7702 | 0.8543 | 0.7916 | 0.8638 | 0.8465 |
70
+ | 0.4871 | 2.0161 | 6000 | 0.4527 | 0.8160 | 0.7475 | 0.8982 | 0.7893 | 0.8665 | 0.8500 |
71
+ | 0.457 | 2.1841 | 6500 | 0.6625 | 0.7653 | 0.6231 | 0.9916 | 0.6838 | 0.8679 | 0.8530 |
72
+ | 0.4685 | 2.3522 | 7000 | 0.5230 | 0.7281 | 0.8393 | 0.6430 | 0.7503 | 0.8681 | 0.8521 |
73
+ | 0.4567 | 2.5202 | 7500 | 0.4433 | 0.8198 | 0.7557 | 0.8956 | 0.7952 | 0.8702 | 0.8533 |
74
+ | 0.4506 | 2.6882 | 8000 | 0.4520 | 0.8208 | 0.7520 | 0.9034 | 0.7948 | 0.8721 | 0.8580 |
75
+ | 0.4443 | 2.8562 | 8500 | 0.4590 | 0.8029 | 0.7963 | 0.8097 | 0.7933 | 0.8730 | 0.8581 |
76
+ | 0.4433 | 3.0242 | 9000 | 0.5848 | 0.7530 | 0.8380 | 0.6837 | 0.7668 | 0.8738 | 0.8592 |
77
+ | 0.4355 | 3.1922 | 9500 | 0.4620 | 0.7974 | 0.8134 | 0.7819 | 0.7933 | 0.8741 | 0.8601 |
78
+
79
+
80
+ ### Framework versions
81
+
82
+ - Transformers 4.42.3
83
+ - Pytorch 2.3.0+cu121
84
+ - Datasets 2.18.0
85
+ - Tokenizers 0.19.0
config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "zhihan1996/DNABERT-2-117M",
3
+ "alibi_starting_size": 512,
4
+ "architectures": [
5
+ "BertForSequenceClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "auto_map": {
9
+ "AutoConfig": "zhihan1996/DNABERT-2-117M--configuration_bert.BertConfig",
10
+ "AutoModel": "zhihan1996/DNABERT-2-117M--bert_layers.BertModel",
11
+ "AutoModelForMaskedLM": "zhihan1996/DNABERT-2-117M--bert_layers.BertForMaskedLM",
12
+ "AutoModelForSequenceClassification": "zhihan1996/DNABERT-2-117M--bert_layers.BertForSequenceClassification"
13
+ },
14
+ "classifier_dropout": null,
15
+ "gradient_checkpointing": false,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.1,
18
+ "hidden_size": 768,
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 3072,
21
+ "layer_norm_eps": 1e-12,
22
+ "max_position_embeddings": 512,
23
+ "model_type": "bert",
24
+ "num_attention_heads": 12,
25
+ "num_hidden_layers": 12,
26
+ "pad_token_id": 0,
27
+ "position_embedding_type": "absolute",
28
+ "problem_type": "single_label_classification",
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.42.3",
31
+ "type_vocab_size": 2,
32
+ "use_cache": true,
33
+ "vocab_size": 4096
34
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32217a75b7d53401288fdbbc64aa7748378a6a4c77230934f1e926e2c3d1d885
3
+ size 356777880
special_tokens_map.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": {
3
+ "content": "[CLS]",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "mask_token": {
10
+ "content": "[MASK]",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "[PAD]",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "sep_token": {
24
+ "content": "[SEP]",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "unk_token": {
31
+ "content": "[UNK]",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ }
37
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[UNK]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "[CLS]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[SEP]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[PAD]",
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
+ "mask_token": "[MASK]",
47
+ "model_max_length": 1000000000000000019884624838656,
48
+ "pad_token": "[PAD]",
49
+ "sep_token": "[SEP]",
50
+ "tokenizer_class": "PreTrainedTokenizerFast",
51
+ "unk_token": "[UNK]"
52
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44d632b1f5466a413eea09ba12767d81365d4907751df1cb88df475912c2c3d0
3
+ size 5176