增加命名空间

def linear_regression():
# 1)准备好数据集:y = 0.8x + 0.7 100个样本
# 特征值X, 目标值y_true
with tf.variable_scope(“original_data”):
X = tf.random_normal(shape=(100, 1), mean=2, stddev=2, name=“original_data_x”)
# y_true [100, 1]
# 矩阵运算 X(100, 1)* (1, 1)= y_true(100, 1)
y_true = tf.matmul(X, [[0.8]], name=“original_matmul”) + 0.7
# 2)建立线性模型:
# y = W·X + b,目标:求出权重W和偏置b
# 3)随机初始化W1和b1
with tf.variable_scope(“linear_model”):
weights = tf.Variable(initial_value=tf.random_normal(shape=(1, 1)), name=“weights”)
bias = tf.Variable(initial_value=tf.random_normal(shape=(1, 1)), name=“bias”)
y_predict = tf.matmul(X, weights, name=“model_matmul”) + bias
# 4)确定损失函数(预测值与真实值之间的误差)-均方误差
with tf.variable_scope(“loss”):
error = tf.reduce_mean(tf.square(y_predict - y_true), name=“error_op”)
# 5)梯度下降优化损失:需要指定学习率(超参数)
# W2 = W1 - 学习率*(方向)
# b2 = b1 - 学习率*(方向)
with tf.variable_scope(“gd_optimizer”):
optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.01, name=“optimizer”).minimize(error)

# 2)收集变量
tf.summary.scalar("error", error)
tf.summary.histogram("weights", weights)
tf.summary.histogram("bias", bias)

# 3)合并变量
merge = tf.summary.merge_all()

# 初始化变量
init = tf.global_variables_initializer()
# 开启会话进行训练
with tf.Session() as sess:
    # 运行初始化变量Op
    sess.run(init)
    print("随机初始化的权重为%f, 偏置为%f" % (weights.eval(), bias.eval()))
    # 1)创建事件文件
    file_writer = tf.summary.FileWriter(logdir="./summary", graph=sess.graph)
    # 训练模型
    for i in range(100):
        sess.run(optimizer)
        print("第%d步的误差为%f,权重为%f, 偏置为%f" % (i, error.eval(), weights.eval(), bias.eval()))
        # 4)运行合并变量op
        summary = sess.run(merge)
        file_writer.add_summary(summary, i)

return None
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值