max_index = -1
candidate = []
candidate_value = []
for j in xrange(len(result[i])):
if result[i][j] > 0.8:
candidate.append(DataParser.label_index[j])
candidate_value.append(result[i][j])
if result[i][j] > max_value:
max_value = result[i][j]
max_index = j
with open('candidate.txt', 'a') as cf:
cf.write(DataParser.label_index[max_index] + ' of ' + str(candidate) + ': ' + str(candidate_value) + '\n')
#f.write(random.choice(candidate) + '\n')
f.write(DataParser.label_index[max_index] + '\n')
correct, all_record = DataParser.check('result.txt', label_file)
print 'correction rate: %i/ %i' %(correct, all_record)
solution_data, solution_id = DataParser.load_test_data(test_file)
solution = test(solution_data)
solution = list(solution)
solution = map(list, zip(*solution)) # transpose
with open('solution.csv', 'w') as f:
f.write('Id,Prediction\n')
for i in xrange(len(solution)):
f.write(solution_id[i] + ',')
max_value = 0
max_index = -1
for j in xrange(len(solution[i])):
if solution[i][j] > max_value:
max_value = solution[i][j]
