作为输入 a 具有浮点数 1.0 或 0.0。当我尝试用我的模型和sparse_categorical_crossentropy损失我得到类似的东西：[[0.4846592 0.5153408]].
我如何知道它预测什么类别？
您看到的这些数字是给定输入样本的每个类别的概率。例如，[[0.4846592 0.5153408]]表示给定样本属于 0 类的概率约为 0.48，属于 1 类的概率约为 0.51。所以你想参加概率最高的课程，因此你可以使用np.argmax https://docs.scipy.org/doc/numpy-1.15.1/reference/generated/numpy.argmax.html查找哪个索引（即 0 或 1）是最大的索引：
import numpy as np
pred_class = np.argmax(probs, axis=-1)
此外，这与模型的损失函数无关。这些概率由模型的最后一层给出，很可能使用softmax作为将输出标准化为概率分布的激活函数。
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
