output updatembk

1 files changed, 6 insertions, 1 deletions

diff --git a/train.cpp b/train.cpp
index a0c62a9..bae88f3 100644
--- a/train.cpp
+++ b/train.cpp
@@ -27,6 +27,7 @@ int train(DataProvider &dp) {
     LOG(INFO)<<"Training started";
     dp.getAllDataSet(D);
     LOG(INFO)<<"Read "<<D.getSize()<<" entries with "<< D.getfSize()<<" features";
+    LOG(INFO)<<"C: "<<C;
     rsvm->train(D);
     vector<double> L;
     rsvm->predict(D,L);
@@ -111,7 +112,8 @@ int main(int argc, char **argv) {
             ("debug,d", "show debug messages")
             ("model,m", po::value<string>(), "set input model file")
             ("output,o", po::value<string>(), "set output model/prediction file")
-            ("feature,i", po::value<string>(), "set input feature file");
+            ("feature,i", po::value<string>(), "set input feature file")
+            ("c,c",po::value<double>(),"trades margin size against training error");
 
     // Parsing program options
     po::store(po::parse_command_line(argc, argv, desc), vm);
@@ -130,6 +132,9 @@ int main(int argc, char **argv) {
 
     mainFunc mainf;
     if (vm.count("train")) {
+        if (vm.count("c")) {
+            C=vm["c"].as<double>();
+        }
         mainf = &train;
     }
     else if (vm.count("validate")||vm.count("predict")) {