keras中的“ predict”和“ predict_class”函数有什么区别?

时间:2018-07-17 13:22:37

标签: keras

keras中的predictpredict_class函数之间有什么区别?

为什么Model对象没有predict_class功能?

2 个答案:

答案 0 :(得分:7)

precdict将返回回归得分,而predict_class将返回您的预测类别。尽管看起来很相似,但还是有一些区别:

想象一下,您正在尝试预测图片是狗还是猫(您有分类器):

predict将返回您:例如0.6只猫和0.2只狗。

predict_class会把你送回猫咪

现在,您正在尝试预测房价(您有一个回归指标):

predict将返回预测价格

predict_class在这里没有意义,因为您没有分类器

TL:DR:对分类器使用predict_class(输出是标签),对回归使用predict(输出不是离散的)

希望有帮助!

答案 1 :(得分:0)

predict_classes 方法仅适用于 Sequential类,而不适用于Model类 You can check this answer