python - 我应该如何解释 sparse_categorical_crossentropy 函数的输出？

Question

作为输入，a 有一个浮点数 1.0 或 0.0。当我尝试使用我的模型和 sparse_categorical_crossentropy 进行预测时损失我得到类似的东西:[[0.4846592 0.5153408]] .

我怎么知道它预测的是什么类别？

Answer 1

您看到的这些数字是给定输入样本的每个类别的概率。例如，[[0.4846592 0.5153408]]表示给定样本属于第 0 类的概率约为 0.48，属于第 1 类的概率约为 0.51。所以你想以最高的概率参加类(class)，因此你可以使用 np.argmax 找出哪个索引(即 0 或 1)是最大值:

import numpy as np

pred_class = np.argmax(probs, axis=-1) 

此外，这与模型的损失函数无关。这些概率由模型中的最后一层给出，很可能它使用了 softmax作为激活函数将输出归一化为概率分布。