0%

线性回归

线性回归

模型

W:权重/b:偏置参数

1
2
3
b = tf.Variable(tf.random_normal([1]), name="bias")
W = tf.Variable(tf.random_normal([1]), name='weight')
Ypred = tf.add(tf.multiply(W, X), b)

损失函数

1
loss = tf.reduce_sum(tf.pow(Ypred - Y, 2)) / (2 * n_samples)

梯度下降算法

1
optimizer = tf.train.AdamOptimizer(learning_rate)

训练示例

数据:美国房价(下载链接)

训练结果:

屏幕快照 2018-07-05 15.52.42

屏幕快照 2018-07-05 15.52.54