post.py 文件源码

def post_ffm(inx):
    out = inx.replace('.csv','_sub.csv')
    idx = "comps/mobike/sol_carl/data/va_20-24.id" 
    last = ''
    pred = {}
    f = open(inx)
    fo = open(out,'w')
    for c,row in enumerate(csv.DictReader(open(idx))):
        line = f.readline()
        row['prob'] = line.strip()
        if last != '' and row['orderid'] != last:
            pred = ','.join(sort_value(pred)[:3])
            fo.write('%s,%s\n'%(last,pred))
            pred = {}
        yp = float(row['prob'])
        pred[row['candidate_loc']] = yp
        last = row['orderid']
        if  c % 10000000 == 0 and c>0:
            print c#, 'log loss', score/(c+1), 'm12 apk', apks/dc
    pred = ','.join(sort_value(pred)[:3])
    fo.write('%s,%s\n'%(row['orderid'],pred))
    fo.close()
    f.close()