5.5 构建模型
5.5.1 定义训练数据占位符
x = tf.placeholder(tf.float32, [None,12], name = "X") # 12个特征数据(12列)
y = tf.placeholder(tf.float32, [None,1], name = "Y") # 1个标签数据(1列)5.5.2 定义模型结构
# 定义了一个命名空间
with tf.name_scope("Model"):
# w 初始化值为shape=(12,1)的随机数
w = tf.Variable(tf.random_normal([12,1], stddev=0.01), name="W")
# b 初始化值为 1.0
b = tf.Variable(1.0, name="b")
# w和x是矩阵相乘,用matmul,不能用mutiply或者*
def model(x, w, b):
return tf.matmul(x, w) + b
# 预测计算操作,前向计算节点
pred= model(x, w, b)
Last updated
Was this helpful?