tf.equal(x, y)
判断x, y是否相等,相等返回true,不相等false
假设x为预测的结果,y为正确的标签值。
import tensorflow as tf
x = [1, 3, 0, 2]
y = [1, 4, 2, 2]
equal = tf.equal(x, y)
with tf.Session() as sess:
print(sess.run(equal))
输出:
[ True False False True]
tf.cast(x, dtype, name=None)
数据类型转换
- x:待转换的数据
- dtype:目标数据类型
- name=None:操作的名称
import tensorflow as tf
x = [1, 3, 0, 2]
y = [1, 4, 2, 2]
equal = tf.equal(x, y)
type = tf.cast(equal, tf.float32)
with tf.Session() as sess:
print(sess.run(type))
输出:
[1. 0. 0. 1.]
tf.reduce_mean()
求均值,上面的代码最后一行改为:
print(sess.run(tf.reduce_mean(type)))
将会输出0.5