-
Notifications
You must be signed in to change notification settings - Fork 1
/
parametrs.py
57 lines (48 loc) · 1.82 KB
/
parametrs.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
max_gradient_norm = 5.0
lr = 0.001
epoch_num = 120
class DialyDialogParams():
embed_size = 100
layer_num = 2
batch_size = 64
hidden_size = 512
source_sequence_length = 32
decoder_length = 45
dataset_path = 'datasets/DailyDialog'
all_questions = dataset_path + '/Q_all.txt'
all_answers = dataset_path + '/A_all.txt'
train_questions = dataset_path + '/train/Q_train.txt'
train_questions_csv = dataset_path + '/train/Q_train.csv'
train_answers = dataset_path + '/train/A_train.txt'
train_answers_csv = dataset_path + '/train/A_train.csv'
validation_questions = dataset_path + '/validation/Q_validation.txt'
validation_questions_csv = dataset_path + '/validation/Q_validation.csv'
validation_answers_csv = dataset_path + '/validation/A_validation.csv'
validation_answers = dataset_path + '/validation/A_validation.txt'
test_questions = dataset_path + '/test/Q_test.txt'
test_questions_csv = dataset_path + '/test/Q_test.csv'
test_answers_csv = dataset_path + '/test/A_test.csv'
test_answers = dataset_path + '/test/A_test.txt'
vocab_path = dataset_path + '/vocab.csv'
model_file_name = '\model.ckpt'
config_file = '\config.json'
class DialogQAParams():
embed_size = 100
layer_num = 2
batch_size = 128
hidden_size = 512
source_sequence_length = 68
decoder_length = 69
dataset_path = 'datasets/DialogQA/pairs'
train_questions = dataset_path + '/src.txt'
train_answers = dataset_path + '/dest.txt'
train_questions_csv = dataset_path + '/src.csv'
train_answers_csv = dataset_path + '/dest.csv'
vocab_path = dataset_path + '/vocab.csv'
model_file_name = '\model.ckpt'
config_file = '\config.json'
# log_file_name = '\logs.txt'
unknown_token = '<unk>'
empty_token = '<EMP>'
start_token = '<s>'
end_token = '</s>'