Repository navigation
Expand file tree
/
Copy pathtransition_cmd.py
More file actions
43 lines (39 loc) · 3.27 KB
/
Copy pathtransition_cmd.py
File metadata and controls
43 lines (39 loc) · 3.27 KB
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
from optparse import OptionParser
def get_parser():
parser = OptionParser()
parser.add_option("--train", dest="conll_train", help="Annotated CONLL train file", metavar="FILE", default="../data/PTB_SD_3_3_0/train.conll")
parser.add_option("--dev", dest="conll_dev", help="Annotated CONLL dev file", metavar="FILE", default="../data/PTB_SD_3_3_0/dev.conll")
parser.add_option("--test", dest="conll_test", help="Annotated CONLL test file", metavar="FILE", default="../data/PTB_SD_3_3_0/test.conll")
parser.add_option("--params", dest="params", help="Parameters file", metavar="FILE", default="params.pickle")
parser.add_option("--extrn", dest="external_embedding", help="External embeddings", metavar="FILE")
parser.add_option("--model", dest="model", help="Load/Save model file", metavar="FILE", default="barchybrid.model")
parser.add_option("--wembedding", type="int", dest="wembedding_dims", default=100)
parser.add_option("--pembedding", type="int", dest="pembedding_dims", default=25)
parser.add_option("--rembedding", type="int", dest="rembedding_dims", default=25)
parser.add_option("--epochs", type="int", dest="epochs", default=30)
parser.add_option("--hidden", type="int", dest="hidden_units", default=100)
parser.add_option("--hidden2", type="int", dest="hidden2_units", default=0)
parser.add_option("--k", type="int", dest="window", default=3)
parser.add_option("--lr", type="float", dest="learning_rate", default=0.1)
parser.add_option("--outdir", type="string", dest="output", default="results")
parser.add_option("--activation", type="string", dest="activation", default="tanh")
parser.add_option("--lstmlayers", type="int", dest="lstm_layers", default=2)
parser.add_option("--lstmdims", type="int", dest="lstm_dims", default=200)
parser.add_option("--dynet-seed", type="int", dest="seed", default=7)
parser.add_option("--disableoracle", action="store_false", dest="oracle", default=True)
parser.add_option("--usehead", action="store_true", dest="headFlag", default=False)
parser.add_option("--userlmost", action="store_true", dest="rlMostFlag", default=False)
parser.add_option("--userl", action="store_true", dest="rlFlag", default=False)
parser.add_option("--predict", action="store_true", dest="predictFlag", default=False)
parser.add_option("--dynet-mem", type="int", dest="cnn_mem", default=512)
parser.add_option("--no-relation", action="store_false", dest="relation", default=True)
parser.add_option("--max-model", type="int", dest="max_model", default=3)
parser.add_option("--action-file", type="string", dest="action_file", default=None)
parser.add_option("--title", type="string", dest="title", default="Default")
parser.add_option("--disable-full-search", action="store_false", dest="full_search", default=True)
parser.add_option("--enable-full-search", action="store_true", dest="full_search", default=True)
parser.add_option("--loss-aug", type="float", dest="loss_aug", default=0.2)
parser.add_option("--disable-beam-search", action="store_false", dest="beam_search", default=True)
parser.add_option("--enable-beam-search", action="store_true", dest="beam_search", default=True)
parser.add_option("--beam-size", type="int", dest="beam_size", default=32)
return parser