Numpy Keras：当使用sigmoid进行二进制分类时，如何从model.predict获取类标签

我在

keras

中实现了一个基本的神经网络，在输出层激活了

sigmoid

。但是，当我对我看不见的数据使用

model.predict（x_new）

时，它会返回一个

numpy

数组，其值范围为

0-1

。我想要一个二进制类输出。如何做到这一点？

您可以使用

predict\u classes（）

方法。或者，您可以将阈值设置为0.5以查找属于类别1的样本：

class\u one=preds>0.5

。感谢您的回复。我使用的是KerasAPI，而不是顺序模型。因此，不能使用预测类方法。因此，唯一的解决方案是手动设置阈值概率。