Skip to content

Instantly share code, notes, and snippets.

@siakon89

siakon89/predict_submit_kaggle_mnist.py

Created July 9, 2018 12:54
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save siakon89/280eac6a446ce6815a3896eca9ec5283 to your computer and use it in GitHub Desktop.
Save siakon89/280eac6a446ce6815a3896eca9ec5283 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
predict_submit_kaggle_mnist.py
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