diff options
Diffstat (limited to 'training/dtrain/dtrain.h')
-rw-r--r-- | training/dtrain/dtrain.h | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/training/dtrain/dtrain.h b/training/dtrain/dtrain.h index b07edfdf..ce5b2101 100644 --- a/training/dtrain/dtrain.h +++ b/training/dtrain/dtrain.h @@ -57,11 +57,18 @@ dtrain_init(int argc, "learning rate [only meaningful if margin>0 or input weights are given]") ("l1_reg,r", po::value<weight_t>()->default_value(0.), "l1 regularization strength [see Tsuruoka, Tsujii and Ananiadou (2009)]") + ("adadelta,D", po::bool_switch()->default_value(false), + "use AdaDelta dynamic learning rates") + ("adadelta_decay", po::value<weight_t>()->default_value(0.9), + "decay for AdaDelta algorithm") + ("adadelta_input", po::value<string>()->default_value(""), + "input for AdaDelta's parameters, two files: file.gradient, and file.update") + ("adadelta_output", po::value<string>()->default_value(""), + "prefix for outputting AdaDelta's parameters") ("margin,m", po::value<weight_t>()->default_value(1.0), "margin for margin perceptron [set =0 for standard perceptron]") ("cut,u", po::value<weight_t>()->default_value(0.1), - "use top/bottom 10% (default) of k-best as 'good' and 'bad' for \ -pair sampling, 0 to use all pairs TODO") + "use top/bottom 10% (default) of k-best as 'good' and 'bad' for pair sampling, 0 to use all pairs TODO") ("adjust,A", po::bool_switch()->default_value(false), "adjust cut for optimal pos. in k-best to cut") ("score,s", po::value<string>()->default_value("nakov"), @@ -87,6 +94,8 @@ pair sampling, 0 to use all pairs TODO") ("max_pairs", po::value<size_t>()->default_value(numeric_limits<size_t>::max()), "max. number of updates/pairs") + ("batch,B", po::bool_switch()->default_value(false), + "perform batch updates") ("output,o", po::value<string>()->default_value("-"), "output weights file, '-' for STDOUT") ("disable_learning,X", po::bool_switch()->default_value(false), @@ -95,6 +104,8 @@ pair sampling, 0 to use all pairs TODO") "output updates (diff. vectors) [to filename]") ("output_raw,R", po::value<string>()->default_value(""), "output raw data (e.g. k-best lists) [to filename]") + ("stop_after", po::value<size_t>()->default_value(numeric_limits<size_t>::max()), + "only look at this number of segments") ("print_weights,P", po::value<string>()->default_value("EgivenFCoherent SampleCountF CountEF MaxLexFgivenE MaxLexEgivenF IsSingletonF IsSingletonFE Glue WordPenalty PassThrough LanguageModel LanguageModel_OOV"), "list of weights to print after each iteration"); po::options_description clopts("Command Line Options"); |