-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
25 lines (21 loc) · 1.17 KB
/
config.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
DEVICE = 3
LOCAL_TEMP_FOLDER = 'temp'
DELETE_TEMP_FILES = True
MODEL_PATHS = {
'token-single_oov': { 'model': 'data/token.char_lstm.ft_oov_tok.44_seed.146.model',
'dset': 'data/token.char_lstm.ft_oov_tok.44_seed.dset'},
'token-single': { 'model': 'data/token.char_cnn.ft_tok.46_seed.104.model',
'dset': 'data/token.char_cnn.ft_tok.46_seed.dset'},
'token-multi_oov': { 'model': 'data/multitok.char_cnn.ft_oov_tok.48_seed.123.model',
'dset': 'data/multitok.char_cnn.ft_oov_tok.48_seed.dset'},
'token-multi': {'model': 'data/multitok.char_cnn.ft_tok.52_seed.173.model',
'dset': 'data/multitok.char_cnn.ft_tok.52_seed.dset'},
'morph_oov': {'model': 'data/morph.char_cnn.ft_oov_yap.49_seed.87.model',
'dset': 'data/morph.char_cnn.ft_oov_yap.49_seed.dset'},
'morph': { 'model': 'data/morph.char_cnn.ft_yap.50_seed.80.model',
'dset': 'data/morph.char_cnn.ft_yap.50_seed.dset'}
}
MULTI_MODEL_FOR_HYBRID = 'token-multi'
YAP_PATH = '../yapproj/src/yap/yap'
YAP_API_URL = 'http://localhost:8000'
YAP_API_HEADERS = {'content-type': 'application/json'}