在使用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)
阈值需要自己选择一下。