深度学习:tensorflow入门:用tf实现线性回归

import tensorflow as tf
import os
os.environ['TP_CPP_MIN_LOG_LEVEL'] = '2'

def myregressin():
    """
    自实现一个线性回归预测
    :return: None
    """
    # 1、准备数据,x 特征值[100, 10]   y 目标值[100]
    x = tf.random_normal([100, 1], mean=1.75, stddev=0.5, name="x_data")

    # 矩阵相乘必须是二维的
    y_true = tf.matmul(x, [[0.7]]) + 0.8

    # 2、建立线性回归模型 1个特征 1个权重 一个偏置 y = wx + b
    # 随机给一个权重和偏置的值,让他去计算损失,然后在当前状态下优化
    # 用变量定义才能优化
    weight = tf.Variable(tf.random_normal([1, 1], mean=0.0, stddev=1.0), name="w")
    bias = tf.Variable(0.0, name="b")

    y_predict = tf.matmul(x, weight) + bias

    # 3、建立损失函数,均方误差
    loss = tf.reduce_mean(tf.square(y_true - y_predict))

    # 4、梯度下降优化损失 learning_rate:0~1,2,3,5,7,10
    train_op = tf.train.GradientDescentOptimizer(0.1).minimize(loss)

    # 定义一个初始化变量的op

    init_op = tf.global_variables_initializer()

    # 通过会话运行程序
    with tf.Session() as sess:
        # 初始化变量
        sess.run(init_op)

        # 打印随机最先初始化的权重和偏置
        print("随机初始化的参数权重为:%f, 偏置为: %f" %(weight.eval(), bias.eval()))

        # 循环训练 运行优化
        for i in range(200):

            sess.run(train_op)
            print("第%d次优化参数权重为:%f, 偏置为: %f" % (i, weight.eval(), bias.eval()))

    return None

if __name__ == '__main__':
    myregressin()
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值