Tensorflow2疑难问题---3、大于某个值为1，小于某个值为0

Tensorflow2疑难问题---3、大于某个值为1，小于某个值为0

一、总结

一句话总结：

可用tf.where方法，tf.where(pridict_y>0.5,x=1,y=0)表示大于0.5取1，否则取0

二、大于某个值为1，小于某个值为0

博客对应课程的视频位置：3、大于某个值为1，小于某个值为0-范仁义-读书编程笔记
https://www.fanrenyi.com/video/37/359

test_y=np.array(test_y)
pridict_y=pridict_y.flatten()
print(test_y)
print(pridict_y)

[0 1 0 1 1 1 1 0 1 1 1 1 1 1 0 0 0 0 1 1 0 1 1 0 0 0 1 1 1 0 1 1 0 0 1 1 0
 0 0 1 0 0 1 1 1 1 1 0 1 0 0 1 0]
[3.15467660e-05 2.46460462e-04 5.71233451e-01 1.94287628e-01
 1.00000000e+00 1.00000000e+00 1.00000000e+00 3.29031190e-31
 1.00000000e+00 1.00000000e+00 1.00000000e+00 1.00000000e+00
 1.00000000e+00 3.40764225e-01 0.00000000e+00 1.22705101e-28
 1.83029751e-08 2.88408506e-03 1.00000000e+00 8.64930868e-01
 1.49744490e-20 1.03670016e-01 1.00000000e+00 3.54205917e-25
 1.04561528e-04 1.75256490e-16 1.00000000e+00 1.00000000e+00
 8.52496326e-01 9.68476042e-15 1.00000000e+00 1.00000000e+00
 0.00000000e+00 1.00000000e+00 3.60370874e-02 1.00000000e+00
 2.00627595e-02 6.50341094e-01 1.00000000e+00 1.00000000e+00
 4.79241669e-01 9.99998927e-01 1.00000000e+00 1.00000000e+00
 2.92697430e-01 1.00000000e+00 3.54994565e-01 0.00000000e+00
 1.00000000e+00 2.37373175e-16 1.00000000e+00 1.00000000e+00
 1.00000000e+00]

需求：

tensorflow 大于某个值为1,小于某个值为0

In [21]:

print(tf.where(pridict_y>0.5,x=1,y=0))
print(test_y)

tf.Tensor(
[0 0 1 0 1 1 1 0 1 1 1 1 1 0 0 0 0 0 1 1 0 0 1 0 0 0 1 1 1 0 1 1 0 1 0 1 0
 1 1 1 0 1 1 1 0 1 0 0 1 0 1 1 1], shape=(53,), dtype=int32)
[0 1 0 1 1 1 1 0 1 1 1 1 1 1 0 0 0 0 1 1 0 1 1 0 0 0 1 1 1 0 1 1 0 0 1 1 0
 0 0 1 0 0 1 1 1 1 1 0 1 0 0 1 0]

In [ ]: