此页面由 Cloud Translation API 翻译。
Switch to English

tf.keras.metrics.categorical_accuracy

TensorFlow 1版 GitHub上查看源代码

计算预测多久一个热标签相匹配。

独立使用:

y_true = [[0,0,1],[0,1,0]] y_pred = [[0.1,0.9,0.8],[0.05,0.95,0]] M = tf.keras.metrics.categorical_accuracy(y_true, y_pred)断言m.shape ==(2,)m.numpy()数组([0,1],D型细胞= FLOAT32)

您可以提供的类logits作为y_pred ,因为logits和概率的argmax相同。

y_true 一热的地面真值。
y_pred 该预测值。

分类准确度值。