Skip to content

Instantly share code, notes, and snippets.

@morrisalp

morrisalp/top_k_categorical_accuracy.py

Created July 24, 2020 07:46
Show Gist options
  • Save morrisalp/a551d6899a372833216a739788b9947f to your computer and use it in GitHub Desktop.
Save morrisalp/a551d6899a372833216a739788b9947f to your computer and use it in GitHub Desktop.
Download ZIP
top K categorical accuracy for numpy arrays (sklearn predict_proba outputs)
Raw
top_k_categorical_accuracy.py
import numpy as np
def top_k_categorical_accuracy(y_true, y_pred_proba, k=1):
return np.equal(np.argsort(y_pred_proba)[:, -k:], y_true[:, None]).any(axis=1).mean()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment