From f35c51af17a306603a1b42e2d182f9b1c3f5151a Mon Sep 17 00:00:00 2001
From: Patrick Simianer <p@simianer.de>
Date: Thu, 16 Jul 2015 13:10:08 +0200
Subject: fixed conf. options

---
 training/dtrain/dtrain.cc | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'training/dtrain/dtrain.cc')

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;
-- 
cgit v1.2.3