gen_result.py 1.1 KB
Newer Older
C
chengxingyi 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
res = []
with open('./rank-00000') as f:
    for line in f:
        pred = map(int,line.strip('\r\n;').split(";"))
        #raw prediction range from 0 to 3
        res.append([i+1 for i in pred])

file_name = open('./data/pred.list').read().strip('\r\n')

FORECASTING_NUM=24
header=['id',
        '201604200805',
        '201604200810',
        '201604200815',
        '201604200820',
        '201604200825',
        '201604200830',
        '201604200835',
        '201604200840',
        '201604200845',
        '201604200850',
        '201604200855',
        '201604200900',
        '201604200905',
        '201604200910',
        '201604200915',
        '201604200920',
        '201604200925',
        '201604200930',
        '201604200935',
        '201604200940',
        '201604200945',
        '201604200950',
        '201604200955',
        '201604201000',
        ]
###################
## To CSV format ##
###################
with open(file_name) as f:
    f.next()
    print ','.join(header)
    for row_num, line in enumerate(f):
        fields = line.rstrip('\r\n').split(',')
        linkid = fields[0]
        print linkid+','+','.join(map(str,res[row_num]))