Christina Theodoris commited on
Commit
1786b44
1 Parent(s): abdf980

Add fine-tuned CellClassifier for cardiomyopathy

Browse files
fine_tuned_models/geneformer-6L-30M_CellClassifier_cardiomyopathies_220224/config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/n/home01/ctheodoris/models/210602_111318_geneformer_27M_L6_emb256_SL2048_E3_B12_LR0.001_LSlinear_WU10000_Oadamw_DS12/models/",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.02,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "relu",
9
+ "hidden_dropout_prob": 0.02,
10
+ "hidden_size": 256,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 512,
18
+ "label2id": {
19
+ "LABEL_0": 0,
20
+ "LABEL_1": 1,
21
+ "LABEL_2": 2
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "max_position_embeddings": 2048,
25
+ "model_type": "bert",
26
+ "num_attention_heads": 4,
27
+ "num_hidden_layers": 6,
28
+ "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
30
+ "problem_type": "single_label_classification",
31
+ "transformers_version": "4.6.0",
32
+ "type_vocab_size": 2,
33
+ "use_cache": true,
34
+ "vocab_size": 25426
35
+ }
fine_tuned_models/geneformer-6L-30M_CellClassifier_cardiomyopathies_220224/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ced328122d57a847fc3914732337674500e259a82e64437c67b4954ac2f4e07
3
+ size 73720721
fine_tuned_models/geneformer-6L-30M_CellClassifier_cardiomyopathies_220224/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:12ead3bad8cf4b853bac87eadeb79c9308ae492e9d29f32da1a2c85e8586108d
3
+ size 41115113
fine_tuned_models/geneformer-6L-30M_CellClassifier_cardiomyopathies_220224/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd8c0a739c2fe6a9ab4bb8f4a62ad8d7b879efcdceb5376b128a2040ff1bbe62
3
+ size 14657
fine_tuned_models/geneformer-6L-30M_CellClassifier_cardiomyopathies_220224/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d0797845afdae765a74ddab7966e0e1837617fd8171af8ee6aef9dedce248f2
3
+ size 623
fine_tuned_models/geneformer-6L-30M_CellClassifier_cardiomyopathies_220224/trainer_state.json ADDED
@@ -0,0 +1,150 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.39658036828041077,
3
+ "best_model_checkpoint": "/n/holyscratch01/xiaoleliu_lab/Users/ctheodoris/models/220224_geneformer_27M_SequenceClassifier_tuning_hCMdCM_L2048_B12_LR1e-05_LScosine_WU500_E1_Oadamw_F2/run-8429a330/checkpoint-7020",
4
+ "epoch": 0.9,
5
+ "global_step": 7020,
6
+ "is_hyper_param_search": true,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.1,
12
+ "learning_rate": 0.00034606438343856935,
13
+ "loss": 0.911,
14
+ "step": 780
15
+ },
16
+ {
17
+ "epoch": 0.1,
18
+ "eval_accuracy": 0.4531576503366612,
19
+ "eval_loss": 1.4550466537475586,
20
+ "eval_runtime": 66.5164,
21
+ "eval_samples_per_second": 259.004,
22
+ "step": 780
23
+ },
24
+ {
25
+ "epoch": 0.2,
26
+ "learning_rate": 0.0006921287668771387,
27
+ "loss": 0.6273,
28
+ "step": 1560
29
+ },
30
+ {
31
+ "epoch": 0.2,
32
+ "eval_accuracy": 0.5953680055723242,
33
+ "eval_loss": 0.846651554107666,
34
+ "eval_runtime": 66.1267,
35
+ "eval_samples_per_second": 260.53,
36
+ "step": 1560
37
+ },
38
+ {
39
+ "epoch": 0.3,
40
+ "learning_rate": 0.0007330550166223805,
41
+ "loss": 0.5592,
42
+ "step": 2340
43
+ },
44
+ {
45
+ "epoch": 0.3,
46
+ "eval_accuracy": 0.5935105641978176,
47
+ "eval_loss": 1.0599186420440674,
48
+ "eval_runtime": 66.2608,
49
+ "eval_samples_per_second": 260.003,
50
+ "step": 2340
51
+ },
52
+ {
53
+ "epoch": 0.4,
54
+ "learning_rate": 0.0006283471571048975,
55
+ "loss": 0.3714,
56
+ "step": 3120
57
+ },
58
+ {
59
+ "epoch": 0.4,
60
+ "eval_accuracy": 0.686324587880195,
61
+ "eval_loss": 1.184874415397644,
62
+ "eval_runtime": 66.1411,
63
+ "eval_samples_per_second": 260.473,
64
+ "step": 3120
65
+ },
66
+ {
67
+ "epoch": 0.5,
68
+ "learning_rate": 0.0005236392975874146,
69
+ "loss": 0.2976,
70
+ "step": 3900
71
+ },
72
+ {
73
+ "epoch": 0.5,
74
+ "eval_accuracy": 0.7681100534014396,
75
+ "eval_loss": 0.6318939328193665,
76
+ "eval_runtime": 66.3309,
77
+ "eval_samples_per_second": 259.728,
78
+ "step": 3900
79
+ },
80
+ {
81
+ "epoch": 0.6,
82
+ "learning_rate": 0.0004189314380699318,
83
+ "loss": 0.2564,
84
+ "step": 4680
85
+ },
86
+ {
87
+ "epoch": 0.6,
88
+ "eval_accuracy": 0.7807058277223126,
89
+ "eval_loss": 0.7283642888069153,
90
+ "eval_runtime": 66.3416,
91
+ "eval_samples_per_second": 259.686,
92
+ "step": 4680
93
+ },
94
+ {
95
+ "epoch": 0.7,
96
+ "learning_rate": 0.0003142235785524487,
97
+ "loss": 0.2336,
98
+ "step": 5460
99
+ },
100
+ {
101
+ "epoch": 0.7,
102
+ "eval_accuracy": 0.8563965637334572,
103
+ "eval_loss": 0.5184123516082764,
104
+ "eval_runtime": 66.3416,
105
+ "eval_samples_per_second": 259.686,
106
+ "step": 5460
107
+ },
108
+ {
109
+ "epoch": 0.8,
110
+ "learning_rate": 0.0002095157190349659,
111
+ "loss": 0.1731,
112
+ "step": 6240
113
+ },
114
+ {
115
+ "epoch": 0.8,
116
+ "eval_accuracy": 0.8288832133735778,
117
+ "eval_loss": 0.5823884010314941,
118
+ "eval_runtime": 66.1535,
119
+ "eval_samples_per_second": 260.425,
120
+ "step": 6240
121
+ },
122
+ {
123
+ "epoch": 0.9,
124
+ "learning_rate": 0.00010480785951748295,
125
+ "loss": 0.1451,
126
+ "step": 7020
127
+ },
128
+ {
129
+ "epoch": 0.9,
130
+ "eval_accuracy": 0.886812166241003,
131
+ "eval_loss": 0.39658036828041077,
132
+ "eval_runtime": 66.3555,
133
+ "eval_samples_per_second": 259.632,
134
+ "step": 7020
135
+ }
136
+ ],
137
+ "max_steps": 7800,
138
+ "num_train_epochs": 1,
139
+ "total_flos": 0,
140
+ "trial_name": null,
141
+ "trial_params": {
142
+ "learning_rate": 0.0008039341830649843,
143
+ "lr_scheduler_type": "polynomial",
144
+ "num_train_epochs": 1,
145
+ "per_device_train_batch_size": 12,
146
+ "seed": 73.15243080311434,
147
+ "warmup_steps": 1812.6785581609881,
148
+ "weight_decay": 0.2588277764570262
149
+ }
150
+ }
fine_tuned_models/geneformer-6L-30M_CellClassifier_cardiomyopathies_220224/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ffee119596c99b50a422b2f80103f4c44f7e25c2ea0e457fe224bad59f1f955
3
+ size 2607