-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
46 lines (41 loc) · 2.48 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
43
44
45
46
import argparse
def get_args():
argp = argparse.ArgumentParser(description='BIDAF', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
# data direction
argp.add_argument('--train_dir', type=str, default='./dataset/squad_v1.1/')
argp.add_argument('--train_file', type=str, default='train-v1.1.json')
argp.add_argument('--dev_file', type=str, default='dev-v1.1.json')
# argp.add_argument('--glove_dir', type=str, default='./dataset/glove.840B.300d/')
# argp.add_argument('--glove_file', type=str, default='glove.840B.300d.txt')
argp.add_argument('--glove_dir', type=str, default='./dataset/glove.6B/')
argp.add_argument('--glove_file', type=str, default='glove.6B.300d.txt')
argp.add_argument('--glove_dict', type=str, default='glove_dict.pkl')
argp.add_argument('--save_dir', type=str, default='./out')
argp.add_argument('--save_pred', type=str, default='./out/predictions.json')
argp.add_argument('--nltk_download', type=str, default='True')
# data control
argp.add_argument('--is_load', type=str, default='True')
argp.add_argument('--max_ques', type=int, default=60)
argp.add_argument('--max_ques_char', type=int, default=30)
argp.add_argument('--max_cont', type=int, default=300) # original text is 791
argp.add_argument('--max_cont_char', type=int, default=37)
argp.add_argument('--batch_size', type=int, default=15)
argp.add_argument('--char_dim', type=int, default=8)
argp.add_argument('--char_6b', type=int, default=1419)
# argp.add_argument('--max_cont_with_char', type=int, default=3054)
# model control
argp.add_argument('--filter_num', type=list, default=[100])
argp.add_argument('--filter_width', type=list, default=[5])
argp.add_argument('--highway_layers', type=int, default=2)
argp.add_argument('--lstm_layers', type=int, default=1)
argp.add_argument('--model_lstm_layers', type=int, default=2)
argp.add_argument('--dropout', type=float, default=0.2)
argp.add_argument('--lr', type=float, default=0.5)
argp.add_argument('--decay_rate', type=float, default=0.999)
argp.add_argument('--max_to_keep', type=int, default=20)
# train control
argp.add_argument('--epochs', type=int, default=12)
argp.add_argument('--mode', type=str, default='train', choices=['train', 'dev', 'test'])
argp.add_argument('--print_step', type=int, default=100)
argp.add_argument('--loss_diff', type=float, default=0.1)
return argp.parse_args()