added single option

1 files changed, 5 insertions, 0 deletions

diff --git a/train.cpp b/train.cpp
index 0b5b4d4..4b8439a 100644
--- a/train.cpp
+++ b/train.cpp
@@ -105,6 +105,7 @@ int main(int argc, char **argv) {
             ("predict,P", "use model for prediction")
             ("cmc,C", "enable cmc auditing")
             ("debug,d", "show debug messages")
+            ("single,s", "one from a pair")
             ("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")
@@ -126,6 +127,10 @@ int main(int argc, char **argv) {
     el::Loggers::reconfigureLogger("default", defaultConf);
 
     mainFunc mainf;
+    if (vm.count("single"))
+        RidList::single=true;
+    else
+        RidList::single=false;
     if (vm.count("train")) {
         if (vm.count("c")) {
             C=vm["c"].as<double>();