You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Paddle/demo/traffic_prediction/gen_result.py

48 lines
1.1 KiB

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]))