AttributeError: ‘Sequential‘ object has no attribute ‘predict_classes‘

这个问题一直困扰很久
AttributeError: ‘Sequential’ object has no
attribute ‘predict_classes’
解决办法:
Keras非顺序模型没有model.predict_classes()方法如何获取测试数据分类的标签
有时分类任务中,当我们用测试数据测试训练好的模型性能时,希望得到模型输出的标签。此时,若搭建的模型为顺序模型,即利用model=Sequential()顺序添加神经网络层时可以直接调用model.predict_calsses()获取预测的标签。

但对于非顺序模型,不存在predict_calsses()方法,我们只能利用predict()得到预测概率。那么这种情况我们如何获取预测标签呢?

在解决问题期间,看到有的博主使用如下代码解决问题

pred=model.predict(X_test)
pred=np.armax(pred,axis=-1)
但是,np.armax(pred,axis=-1)返回值为最大值的索引,所以只能对多分类模型或者是以softmax作为输出层激活函数的二分类模型

对于输出层激活函数为sigmoid的模型,其输出只有一个数,所以利用上述方法所有输出都会被映射到一个标签,因此,针对这种情况。应该利用

Y_pred = model.predict(X_test)
Y_pred = np.int64(Y_pred>0.5)
已解决

  • 9
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

人工智能小李子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值