ragunath-ravi commited on
Commit
f2bf8af
·
1 Parent(s): 9817e2a

Upload folder using huggingface_hub

Browse files
model/hcat_checkpoints_v_improved/enhanced_hcat_best_avg.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:495b3d1e9404fc2c4f79da73bb09210574e3131bfaa1b335f83ecab433e92116
3
+ size 141785185
model/hcat_checkpoints_v_improved/enhanced_hcat_best_rec.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21b9aee5b97f32043c65cc8db6c7f074eb57a650691e958247681473efc2e6a3
3
+ size 141785185
model/hcat_checkpoints_v_improved/enhanced_hcat_best_surv.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b071505d2543cc71d697745fbeb9d6f1753c511b155cf27228fdbdde0c8fbbb6
3
+ size 141785551
model/hcat_checkpoints_v_improved/enhanced_hcat_final.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50e915e8ba054ed64d8fa9bfc7bf60e27f6339c1f6cca2e4f6d4ddb9c2ff3bdc
3
+ size 141784023
model/hcat_checkpoints_v_improved/enhanced_hcat_training_summary.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "n_patients": 763,
3
+ "best_metrics": {
4
+ "surv_f1": 0.8,
5
+ "rec_f1": 0.95,
6
+ "avg_f1": 0.875
7
+ },
8
+ "history": {
9
+ "train_loss": [
10
+ 1.5406978004857113,
11
+ 0.9772744335626301,
12
+ 0.7430524684880909,
13
+ 0.607304478946485,
14
+ 0.5724760293960571,
15
+ 0.5759610470972563,
16
+ 0.5099517624629172,
17
+ 0.42624059633204814,
18
+ 0.40211578889896993,
19
+ 0.37630057413327067,
20
+ 0.37706336771187027,
21
+ 0.29755159036109324
22
+ ],
23
+ "val_surv_f1": [
24
+ 0.711864406779661,
25
+ 0.7008547008547008,
26
+ 0.7068965517241379,
27
+ 0.7068965517241379,
28
+ 0.780952380952381,
29
+ 0.7787610619469026,
30
+ 0.780952380952381,
31
+ 0.7889908256880734,
32
+ 0.8,
33
+ 0.7636363636363637,
34
+ 0.7350427350427351,
35
+ 0.7747747747747747
36
+ ],
37
+ "val_rec_f1": [
38
+ 0.9230769230769231,
39
+ 0.9230769230769231,
40
+ 0.95,
41
+ 0.95,
42
+ 0.95,
43
+ 0.95,
44
+ 0.95,
45
+ 0.95,
46
+ 0.95,
47
+ 0.95,
48
+ 0.95,
49
+ 0.95
50
+ ],
51
+ "val_avg_f1": [
52
+ 0.8174706649282921,
53
+ 0.811965811965812,
54
+ 0.8284482758620689,
55
+ 0.8284482758620689,
56
+ 0.8654761904761905,
57
+ 0.8643805309734514,
58
+ 0.8654761904761905,
59
+ 0.8694954128440366,
60
+ 0.875,
61
+ 0.8568181818181818,
62
+ 0.8425213675213675,
63
+ 0.8623873873873873
64
+ ]
65
+ },
66
+ "model_config": {
67
+ "d_model": 512,
68
+ "n_heads": 8,
69
+ "n_global_layers": 3,
70
+ "dropout": 0.2,
71
+ "use_advanced_imputation": true,
72
+ "n_impute_iterations": 3
73
+ },
74
+ "training_config": {
75
+ "epochs": 50,
76
+ "batch_size": 32,
77
+ "learning_rate": 5e-05,
78
+ "weight_decay": 0.0001,
79
+ "moddrop": 0.1,
80
+ "emb_noise": 0.05
81
+ }
82
+ }