Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
with open('submission.csv', 'w') as f:
f.write('ImageId,Label\n')
for data in tqdm(test_data):
arr = numpy.expand_dims(data[0], axis=0)
number = model.predict(arr)
label = argmax(number)
f.write(str(data[1]) + ',' + str(label) + '\n')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment