在使用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)
閾值需要自己選擇一下。