Skip to content

Commit 7d15559

Browse files
committed
added modality
1 parent a90c7c3 commit 7d15559

22 files changed

+8963
-1
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
.tar

__pycache__/config.cpython-37.pyc

361 Bytes
Binary file not shown.

modal_clinical.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828
from config import HyperP
2929

30-
hyp = HyperP(model_type = "attn_train_best_config") # slope prediction
30+
hyp = HyperP(model_type = "singlemodal_clinical") # slope prediction
3131

3232
# seed
3333
seed = hyp.seed
@@ -155,6 +155,11 @@ class TabCT(nn.Module):
155155
def __init__(self, cnn, attn_filters, fc_dim, n_attn_layers):
156156
super(TabCT, self).__init__()
157157

158+
self.n_tab = hyp.n_tab # n tabular features
159+
self.attn_filters = attn_filters
160+
self.fc_dim = fc_dim
161+
self.n_attn_layers = n_attn_layers
162+
158163

159164
self.fc_inter = nn.Linear(self.n_tab, self.fc_dim)
160165

0 commit comments

Comments
 (0)