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 [ ]: