使用神经网络训练的监督分类对不同类别的概率估计

时间:2020-06-13 22:35:36

标签: machine-learning keras classification tensorflow2.0 supervised-learning

在教程https://www.tensorflow.org/tutorials/keras/classification

at

https://www.tensorflow.org/tutorials/keras/classification#make_predictions

预测是10个数字组成的数组。它们代表模型的 “确信”该图像对应于10种不同图像中的每一种 衣服。您可以看到哪个标签的最高 置信度值:

如果我想估计每堂课(不同衣服的概率)的话,那就是信心不足。我该怎么办?

1 个答案:

答案 0 :(得分:0)

@desertnaut在上面提到评论时,对代码的信心

probability_model = tf.keras.Sequential([model, 
                                         tf.keras.layers.Softmax()])
predictions = probability_model.predict(test_images)

由变量predictions给出的确实是概率。