-
Notifications
You must be signed in to change notification settings - Fork 10
/
config.py
33 lines (25 loc) · 1.28 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
26
27
28
29
30
31
32
33
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_HOST = 'localhost'
YAP_API_PORT = '8000'
YAP_API_URL_TEMPLATE = 'http://{host}:{port}'
YAP_API_HEADERS = {'content-type': 'application/json'}
MAX_THREADS_FASTAPI = 5
SUPPRESS_IOBES_WARNINGS = True