Skip to content

Instantly share code, notes, and snippets.

@Sanlap1997

Sanlap1997/class tain.py Secret

Created September 24, 2020 06:53
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 Sanlap1997/79a34d5a953dad9f606fd5cffcaf6506 to your computer and use it in GitHub Desktop.
Save Sanlap1997/79a34d5a953dad9f606fd5cffcaf6506 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
class tain.py
Y_pred_test = model.predict_proba(X_test_normalized) # Predict probability of image belonging to a class, for each class
Y_pred_test_classes = np.argmax(Y_pred_test, axis=1) # Class with highest probability from predicted probabilities
Y_test_classes = np.argmax(Y_test_coded, axis=1) # Actual class
Y_pred_test_max_probas = np.max(Y_pred_test, axis=1) # Highest probability
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment