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

hyq*_*dvd 5 keras

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

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

Ign*_*ier 10

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

假设您要预测图片是狗还是猫(您有分类器):

  • predict 将返回您:0.6猫和0.2狗(例如)。
  • predict_class 会把猫还给你

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

  • predict 将返回预测价格
  • predict_class 由于您没有分类器,因此这里没有意义

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

希望能帮助到你!

  • 如果'predict_class'函数用于分类任务,为什么Model对象没有这个函数? (4认同)

Har*_*pta 7

predict_classes方法仅适用于Sequential 类 ,不适用于 Model 类 您可以查看此答案