1 files changed, 3 insertions, 2 deletions
diff --git a/misc.py b/misc.py
index 7bf1370..1a946ed 100644
--- a/misc.py
+++ b/misc.py
@@ -7,12 +7,13 @@ def split(params):
 
 def train(params):
     devnull = open(os.devnull, 'w')
+    params+=" --iter 1 --cg_prec 1e-4 --ls_prec 1e-10"
     call(["./ranksvm"]+params.split(" "),stdout=devnull)
 
 def cmc(params):
     devnull = open(os.devnull, 'w')
-    retcode = check_output(["./ranksvm"]+params.split(" "));
-    return [float(i) for i in retcode.split('\n')[1:]]
+    retcode = check_output(["./ranksvm"]+params.split(" ")).decode('ascii').split('\n');
+    return float(retcode[-2])
 
 def take(fname):
     f=open(fname,'r')