diff options
Diffstat (limited to 'misc.py')
| -rw-r--r-- | misc.py | 31 |
1 files changed, 30 insertions, 1 deletions
@@ -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]] |