final

1 files changed, 7 insertions, 1 deletions

diff --git a/misc.py b/misc.py
index 84ce869..b546ad3 100644
--- a/misc.py
+++ b/misc.py
@@ -6,7 +6,7 @@ def split(params):
     call(["./split"]+params.split(" "),stdout=devnull)
 
 def train(params):
-    params+=" --iter 1 --cg_prec 1e-4 --ls_prec 1e-10 --prec 1e-4"
+    params+=" --iter 1 --cg_prec 1e-4 --ls_prec 1e-10 --prec 1e-4 -M msk"
     bare(params)
 
 def bare(params):
@@ -18,6 +18,12 @@ def cmc(params):
     retcode = check_output(["./ranksvm"]+params.split(" ")).decode('ascii').split('\n');
     return float(retcode[-2])
 
+def cmcc(params):
+    devnull = open(os.devnull, 'w')
+    retcode = check_output(["./ranksvm"]+params.split(" ")).decode('ascii').split('\n');
+    return [float(x) for x in retcode[1:101]]
+
+
 def take(fname):
     f=open(fname,'r')
     res=f.read().split('\n')