-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.py
42 lines (28 loc) · 1.47 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
34
35
36
37
38
39
40
41
42
import os
_CUR_DIR = os.path.dirname(os.path.abspath(__file__))
# ファイルの場所
# ソース
PATH_TO_VGR_domain_text = _CUR_DIR + '/data/VGR_domain_text'
PATH_TO_VGR_domain_text2 = _CUR_DIR + '/data/VGR_domain_text2'
# 単語レベルと文字レベルの語彙
PATH_TO_WORD_VOCAB = _CUR_DIR + '/data/vocab/word_vocab.txt'
PATH_TO_CHAR_VOCAB = _CUR_DIR + '/data/vocab/char_vocab.txt'
PATH_TO_WORD_VOCAB2 = _CUR_DIR + '/data/vocab/word_vocab2.txt'
PATH_TO_CHAR_VOCAB2 = _CUR_DIR + '/data/vocab/char_vocab2.txt'
# 入出力用データ
PATH_TO_X_TRAIN = _CUR_DIR + '/data/train/X_train.txt'
PATH_TO_Y_TRAIN = _CUR_DIR + '/data/train/y_train.txt'
PATH_TO_X_TEST = _CUR_DIR + '/data/test/X_test.txt'
PATH_TO_Y_TEST = _CUR_DIR + '/data/test/y_test.txt'
# 英語データ
PATH_TO_ENG_X_TRAIN = _CUR_DIR + '/data/AESW/AESW_X_train.txt'
PATH_TO_ENG_Y_TRAIN = _CUR_DIR + '/data/AESW/AESW_y_train.txt'
PATH_TO_ENG_X_TEST = _CUR_DIR + '/data/AESW/AESW_X_test.txt'
PATH_TO_ENG_Y_TEST = _CUR_DIR + '/data/AESW/AESW_y_test.txt'
PATH_TO_ENG_SOURCE_WORD_VOCAB = _CUR_DIR + '/data/vocab/eng_source_word_vocab.txt'
PATH_TO_ENG_TARGET_WORD_VOCAB = _CUR_DIR + '/data/vocab/eng_target_word_vocab.txt'
PATH_TO_ENG_SOURCE_CHAR_VOCAB = _CUR_DIR + '/data/vocab/eng_source_char_vocab.txt'
PATH_TO_ENG_TARGET_CHAR_VOCAB = _CUR_DIR + '/data/vocab/eng_target_char_vocab.txt'
PATH_TO_ENG_TARGET_VOCAB = _CUR_DIR + '/data/vocab/eng_target_vocab.txt'
# 学習済みモデル
PATH_TO_MODELS = _CUR_DIR + '/model/'