tf.keras.metrics.sparse_categorical_accuracy

Вычисляет частоту совпадения предсказаний с целочисленными метками.

tf.keras.metrics.sparse_categorical_accuracy(
    y_true, y_pred
)

Использование в отдельном блоке:

y_true = [2, 1]
y_pred = [[0.1, 0.9, 0.8], [0.05, 0.95, 0]]
m = tf.keras.metrics.sparse_categorical_accuracy(y_true, y_pred)
assert m.shape == (2,)
m.numpy()
array([0., 1.], dtype=float32)

Вы можете предоставить логарифмы классов как y_pred, так как argmax логарифмов и вероятностей одинаковы.