-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathconstants.py
117 lines (95 loc) · 5.71 KB
/
constants.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
PAD_CHAR = "**PAD**"
EMBEDDING_SIZE = 100
# MAX_LENGTH = 512
MAX_LENGTH = 2500
# where you want to save any models you may train
# MODEL_DIR = './outputs/bert-512-5e_5-basic/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-basic-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-basic-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-basic-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-SA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-SA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-SA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-CA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-CA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-CA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-CA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-pos/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-CA/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-CA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-CA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-CA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-basic/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-basic-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-basic-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-basic-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-SA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-SA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-SA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-CA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-CA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-CA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-CA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-basic/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-basic-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-basic-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-basic-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-SA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-SA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-SA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-CA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-CA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-CA-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-CA-pos-pretrain/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-pos/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-reTok/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-pos-reTok/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA-reTok/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-SA-pos-reTok/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-basic-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-basic-pos-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-basic-reTok-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/bert-512-5e_5-basic-pos-reTok-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-pos-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-reTok-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/biobert-512-5e_5-basic-pos-reTok-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/bert-small-2500-5e_4-basic/MIMIC3-50'
# MODEL_DIR = './outputs/bert-small-2500-5e_4-basic-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-small-2500-5e_4-basic-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/bert-small-2500-5e_4-basic-pos-scratch/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-B/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-B-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-SA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-CA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-512-5e_5-CA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-reTok-B/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-reTok-B-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-reTok-SA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-reTok-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-reTok-CA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-2500-5e_5-reTok-CA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-CAML-2500-5e_5-B/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-CAML-2500-5e_5-B-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-CAML-2500-5e_5-SA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-CAML-2500-5e_5-SA-pos/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-CAML-2500-5e_5-CA/MIMIC3-50'
# MODEL_DIR = './outputs/bert-tiny-CAML-2500-5e_5-CA-pos/MIMIC3-50'
MODEL_DIR = './outputs/debug3'
DATA_DIR = './mimicdata/'
MIMIC_3_DIR = './mimicdata/mimic3'
MIMIC_2_DIR = './mimicdata/mimic2'