tf.keras.metrics.top_k_categorical_accuracy