欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

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

时间:2021-01-22 12:27:34|栏目:Python代码|点击:

1 predict()方法

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

2 predict_classes()方法

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

补充知识:keras中model.evaluate、model.predict和model.predict_classes的区别

1、model.evaluate 用于评估您训练的模型。它的输出是model的acc和loss,而不是对输入数据的预测。

2、model.predict 实际预测,输入为test sample,输出为label。

3、在keras中有两个预测函数model.predict_classes(test) 和model.predict(test)。如果标签经过了one-hot编码,如[1,2,3,4,5]是标签类别,经编码后为[1 0 0 0 0],[0 1 0 0 0]…[0 0 0 0 1]。

model.predict_classes(test)预测的是类别,打印出来的值就是类别号。并且只能用于序列模型来预测,不能用于函数式模型。

而model.predict(test)输出的还是5个编码值,要经过argmax(predict_test,axis=1)转化为类别号。

上一篇:python实现随机漫步方法和原理

栏    目:Python代码

下一篇:python 远程统计文件代码分享

本文标题:对Keras中predict()方法和predict_classes()方法的区别说明

本文地址:http://www.codeinn.net/misctech/49321.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有