-
Notifications
You must be signed in to change notification settings - Fork 1
/
args.py
20 lines (17 loc) · 1.22 KB
/
args.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import argparse
def process_command():
parser = argparse.ArgumentParser(prog='Training', description='Arguments')
parser.add_argument('--dlen', '-dl', default=300, help='doc len', type=int )
parser.add_argument('--wlen', '-wl', default=30, help='word len', type=int )
parser.add_argument('--wdim', '-wd', default=27, help='word dim', type=int )
parser.add_argument('--hdim', '-hd', default=50, help='hidden dim', type=int )
parser.add_argument('--label', '-l', default=2, help='output label', type=int )
parser.add_argument('--flen', '-fl', default=3, help='filter length', type=int )
parser.add_argument('--channel', '-c', default=50, help='channel size', type=int )
parser.add_argument('--batch', '-b', default=64, help='batch size', type=int )
parser.add_argument('--epoch', '-e', default=25, help='epoch', type=int )
parser.add_argument('--gpu', '-g', default=0, help='-1=cpu, 0, 1,...= gpt', type=int)
parser.add_argument('--model', '-model', default='./model', help='path of model')
parser.add_argument('--train', '-train', default='./data/semeval_wo_stem/train.txt', help='path of training data')
parser.add_argument('--predict', '-predict', default='./data/test.txt', help='path of predicted data')
return parser.parse_args()