对y_pred强制二分类

在使用Keras做0/1二分类问题时,采用sigmoid作为输出层的激活函数。但是结果中并非0和1,而是介于(0,1)之间的小数。找了其他的激活函数,发现没有合适的。

所以直接强制二分类吧。

y_test_pred = fit.predict(x_test)
y_test_pred_binary = []
for item in y_test_pred:
    if item <= 0.5:
        a = 0
    else:
        a = 1
    y_test_pred_binary.append(a)

阈值需要自己选择一下。

猜你喜欢

转载自blog.csdn.net/weixin_41171061/article/details/83861923