您的当前位置:首页正文

Keras中predict()方法和predict_classes()方法的区别

2024-11-07 来源:个人技术集锦

1 predict()方法
        当使用predict()方法进行预测时,返回值是数值,表示样本属于每一个类别的概率,我们可以使用numpy.argmax()方法找到样本以最大概率所属的类别作为样本的预测标签。

2 predict_classes()方法
        当使用predict_classes()方法进行预测时,返回的是类别的索引,即该样本所属的类别标签。以卷积神经网络中的图片分类为例说明,代码如下:

Top