InvalidArgumentError: cannot compute Equal as input #1(zero-based) was expected to be a int32 tensor

InvalidArgumentError: cannot compute Equal as input #1(zero-based) was expected to be a int32 tensor but is a float tensor [Op: Equal] 错误解决!

在这里插入图片描述

在这里插入图片描述
如高亮黄色处,知道Op: Equal是值Equal判断运算出错,tf.equal()函数比较两个元素需要数据类型相同,而上述程序tf.where(PRED.numpy()<0.5,0,1)结果是0或1为int32,而Y类型为float32无法比较,需要将语句tf.where(PRED.numpy()<0.5,0,1)中<0.5后面的0,1改为0. ,1. 变成浮点型。

在这里插入图片描述