Pandas 如何将0/1作为二进制预测输出而不是概率？

我训练了一个神经网络模型，得到了介于

0

和

1

之间的预测结果，如何将它们转换成0/1而不是小数？我设法将它们转换为bolean值：

y_pred

output:
array([[0.05447599],
       [0.09883076],
       [0.11023161],
       ...,
       [0.19569233],
       [0.07266018],
       [0.08473385]], dtype=float32)

y_pred_nn = (y_pred_nn > 0.5)

output:
array([[False],
       [False],
       [False],
       ...,
       [False],
       [False],
       [False]])

预期结果：

array([[0],
       [0],
       [0],
       ...,
       [0],
       [0],
       [0]], dtype=float32)

#那就转换成浮点数吧
y_pred_nn=（y_pred_nn>0.5）.aType（np.32）