summaryrefslogtreecommitdiff
path: root/training/dtrain
diff options
context:
space:
mode:
authorPatrick Simianer <p@simianer.de>2015-07-20 20:09:37 +0200
committerPatrick Simianer <p@simianer.de>2015-07-20 20:09:37 +0200
commite606d71a31038281d141022cd8c26a21cada3f27 (patch)
tree806b3508883e44ee5fbba6798c4ec03f91e3fb73 /training/dtrain
parentf35c51af17a306603a1b42e2d182f9b1c3f5151a (diff)
fix
Diffstat (limited to 'training/dtrain')
-rw-r--r--training/dtrain/dtrain.cc2
-rw-r--r--training/dtrain/dtrain.h6
2 files changed, 5 insertions, 3 deletions
diff --git a/training/dtrain/dtrain.cc b/training/dtrain/dtrain.cc
index 42a771e0..b44f8f2b 100644
--- a/training/dtrain/dtrain.cc
+++ b/training/dtrain/dtrain.cc
@@ -68,12 +68,14 @@ main(int argc, char** argv)
cerr << _p4;
// output configuration
cerr << "Parameters:" << endl;
+ cerr << setw(25) << "bitext " << "'" << input_fn << "'" << endl;
cerr << setw(25) << "k " << k << endl;
cerr << setw(25) << "score " << "'" << score_name << "'" << endl;
cerr << setw(25) << "N " << N << endl;
cerr << setw(25) << "T " << T << endl;
cerr << setw(25) << "learning rate " << eta << endl;
cerr << setw(25) << "margin " << margin << endl;
+ cerr << setw(25) << "average " << average << endl;
cerr << setw(25) << "l1 reg " << l1_reg << endl;
cerr << setw(25) << "decoder conf " << "'"
<< conf["decoder_conf"].as<string>() << "'" << endl;
diff --git a/training/dtrain/dtrain.h b/training/dtrain/dtrain.h
index 1d07defa..cdc2f628 100644
--- a/training/dtrain/dtrain.h
+++ b/training/dtrain/dtrain.h
@@ -65,12 +65,12 @@ dtrain_init(int argc, char** argv, po::variables_map* conf)
"list of weights to print after each iteration");
po::options_description clopts("Command Line Options");
clopts.add_options()
- ("conf,c", po::value<string>(), "dtrain configuration file")
- ("help,h", po::bool_switch()->default_value(false), "display options");
+ ("conf,c", po::value<string>(), "dtrain configuration file")
+ ("help,h", po::bool_switch(), "display options");
opts.add(clopts);
po::store(parse_command_line(argc, argv, opts), *conf);
cerr << "dtrain" << endl << endl;
- if (conf->count("help")) {
+ if ((*conf)["help"].as<bool>()) {
cerr << opts << endl;
return false;