tf.keras.metrics.top_k_categorical_accuracy

tf.keras.metrics.top_k_categorical_accuracy(
    y_true,
    y_pred,
    k=5
)

Defined in tensorflow/python/keras/_impl/keras/metrics.py.