tensorflow写线性回归代码

'''
线性回归
'''

'''
线性回归
'''

import os

import tensorflow.compat.v1 as tf
tf.compat.v1.disable_eager_execution()

x = tf.random_normal([100,1],mean = 1.75, stddev=0.5)
y= tf.matmul(x,[[2.0]])+ 5.0

#构建模型
init_w = tf.random_normal(shape=[1,1])
weight = tf.Variable(init_w,trainable=True)
bias = tf.Variable(0.0, trainable=True)

pred_y =tf.matmul(x, weight) + bias
#损失函数
loss = tf.reduce_mean(tf.square(y-pred_y))
#梯度下降优化器
train_op = tf.train.GradientDescentOptimizer(0.1).minimize(loss)

#定义收集损失函数
tf.summary.scalar('losses', loss)
merged = tf.summary.merge_all()

saver = tf.train.Saver()

#执行
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())

    fw = tf.summary.FileWriter('../summary', graph= sess.graph)
    if os.path.exists('../model/lr/checkpoint'):
        saver.restore(sess, '../model/lr/')
        
    
    for i in range(500):
        sess.run(train_op)
        summary = sess.run(merged)
        fw.add_summary(summary, i)

        print('轮数:{}, w: {}, b: {}'.format(i+1, weight.eval(),bias.eval()))
    saver.save(sess, '../model/lr/')