无论如何,我是否可以将使用if
之后获得的值存储在变量中(以便以后在model.predict_on_batch()
条件中使用)?
例如,我的问题得到了3个输出,而我得到了这个结果:
<tf.Tensor: id=1159525, shape=(1, 3), dtype=float32, numpy=array([[ 2.0149750e+03, 1.2258940e+03, -1.5718406e-01]], dtype=float32)>
所以,就像这样:
y1_pred = 2.0149750e+03
y2_pred = 1.2258940e+03
y3_pred = -1.5718406e-01
if y1_pred < 3000 and y1_true < 3000:
print("Ok!")
else:
print("Not ok!")