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

閾值需要自己選擇一下。

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