Skip to content

Instantly share code, notes, and snippets.

@Praveenk8051

Praveenk8051/model_predict_mnist.py

Created May 6, 2020 11:41
Show Gist options
  • Save Praveenk8051/4902399e3edecd92231ec3866e6a9634 to your computer and use it in GitHub Desktop.
Save Praveenk8051/4902399e3edecd92231ec3866e6a9634 to your computer and use it in GitHub Desktop.
Download ZIP
model prediction
Raw
model_predict_mnist.py
# prediction
results = model.predict(test)
# Index with the maximum probability
results = np.argmax(results,axis = 1)
results = pd.Series(results,name="Label")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment