From b44da2a2ab11425961014a39994484c92626ce58 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Tue, 2 Jun 2015 18:10:33 +0800
Subject: misc

---
 train.cpp | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'train.cpp')

diff --git a/train.cpp b/train.cpp
index fda06cb..dd1b1e6 100644
--- a/train.cpp
+++ b/train.cpp
@@ -72,9 +72,19 @@ int predict(DataProvider &dp) {
             rank_CMC(D,L,cmc);
     }
 
-    if (vm.count("output") && vm.count("predict"))
+    if (vm.count("predict"))
+    {
+        if (vm.count("pair"))
+        {
+            vector<double> pair;
+            rank_pair(D,L,pair);
+            for (int i=0;i<pair.size();++i)
+                *ot<<pair[i]<<endl;
+        }
+        else
         for (int i=0; i<L.size();++i)
             *ot<<L[i]<<endl;
+    }
 
     LOG(INFO)<<"Finished";
     if (vm.count("cmc"))
@@ -110,6 +120,7 @@ int main(int argc, char **argv) {
             ("cmc,C", "enable cmc auditing")
             ("debug,d", "show debug messages")
             ("single,s", "one from a pair")
+            ("pair,p","get pair result")
             ("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")
-- 
cgit v1.2.3-70-g09d2