tf.keras.metrics.TopKCategoricalAccuracy