对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)

阈值需要自己选择一下。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章