From 28af3b75083382c29a56a34dc2f6e0051606e3c7 Mon Sep 17 00:00:00 2001 From: Joe Zhao Date: Wed, 3 Jun 2015 10:44:45 +0800 Subject: ensemble --- misc.py | 31 ++++++++++++++++++++++++++++++- 1 file changed, 30 insertions(+), 1 deletion(-) (limited to 'misc.py') diff --git a/misc.py b/misc.py index 1a946ed..84ce869 100644 --- a/misc.py +++ b/misc.py @@ -6,8 +6,11 @@ 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" + bare(params) + +def bare(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): @@ -29,3 +32,29 @@ def put(fname,a): for item in a: f.write("%s\n" % item) f.close() + +def getmodel(fname): + f=open(fname,'r') + res=f.read().split('\n') + f.close() + mod=[]; + fsize=int(res[1]); + for i in range(fsize): + mod.append(float(res[3+i])) + return mod; + +def putmodel(fname,m): + f=open(fname,'w') + f.write("TN\n") + f.write("%d\n" % len(m)) + f.write("%d 1\n"%len(m)) + for a in m: + f.write("%g\n"%a) + f.write("0") + f.close() + +def getpred(fname): + f=open(fname,'r') + res=f.read().split('\n') + f.close() + return [float(p) for p in res[:-1]] -- cgit v1.2.3-70-g09d2