-
Notifications
You must be signed in to change notification settings - Fork 0
/
configure.py
44 lines (39 loc) · 1.12 KB
/
configure.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
import argparse
import sys
def parse_args():
"""
Parse input arguments
"""
parser = argparse.ArgumentParser()
parser.add_argument(
"--bert_model",
default="bert-base-uncased",
type=str,
help="BERT model. Options: bert-base-uncased/bert-large-uncased, default is 'bert-base-uncased'",
)
parser.add_argument(
"--type_phrases",
default="",
type=str,
help="SP-10K syntactic relation types. Options: nsubj/nsubj_amod/amod/dobj/dobj_amod",
)
parser.add_argument(
"--type_attention",
default="standard",
type=str,
help="standard, head, context, or control",
)
parser.add_argument(
"--corpus_dir",
default="",
type=str,
help="The directory of the corpus to extract sents from.",
)
parser.add_argument(
"--sp_dir", default="", type=str, help="The directory where SP-10K is saved."
)
parser.add_argument(
"--sent_file", default="", type=str, help="The file with the preprocessed sentences."
)
args = parser.parse_args()
return args