tf.keras.metrics.top_k_categorical_accuracy

TensorFlow 2 version View source on GitHub