summaryrefslogtreecommitdiff
path: root/training/dtrain/dtrain.cc
diff options
context:
space:
mode:
authorPatrick Simianer <p@simianer.de>2015-07-16 13:10:08 +0200
committerPatrick Simianer <p@simianer.de>2015-07-16 13:10:08 +0200
commitf35c51af17a306603a1b42e2d182f9b1c3f5151a (patch)
tree7107566db9a52ed1176de83f760771b6482f4f8b /training/dtrain/dtrain.cc
parent1deb94cada467ac0fe171ae4182c37c3d217c17a (diff)
fixed conf. options
Diffstat (limited to 'training/dtrain/dtrain.cc')
-rw-r--r--training/dtrain/dtrain.cc5
1 files changed, 3 insertions, 2 deletions
diff --git a/training/dtrain/dtrain.cc b/training/dtrain/dtrain.cc
index 97df530b..42a771e0 100644
--- a/training/dtrain/dtrain.cc
+++ b/training/dtrain/dtrain.cc
@@ -10,7 +10,8 @@ main(int argc, char** argv)
{
// get configuration
po::variables_map conf;
- dtrain_init(argc, argv, &conf);
+ if (!dtrain_init(argc, argv, &conf))
+ return 1;
const size_t k = conf["k"].as<size_t>();
const string score_name = conf["score"].as<string>();
const size_t N = conf["N"].as<size_t>();
@@ -66,7 +67,7 @@ main(int argc, char** argv)
cerr << _p4;
// output configuration
- cerr << "dtrain" << endl << "Parameters:" << endl;
+ cerr << "Parameters:" << endl;
cerr << setw(25) << "k " << k << endl;
cerr << setw(25) << "score " << "'" << score_name << "'" << endl;
cerr << setw(25) << "N " << N << endl;