From 5297c7d0358f01193f9057fa0fcae82d55a2872b Mon Sep 17 00:00:00 2001 From: Joe Zhao Date: Tue, 19 May 2015 18:09:58 +0800 Subject: added single option --- train.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'train.cpp') 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(), "set input model file") ("output,o", po::value(), "set output model/prediction file") ("feature,i", po::value(), "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(); -- cgit v1.2.3-70-g09d2