初学 Tensorflow(线性回归预测)

大致过程是:

先生成函数为 y = 0.1x + 0.3 的x、y的数据,即 k = 0.1, b = 0.3

设 k, b 初始为随机数,进行 200 次学习,每隔 20 次输出一下训练结果


具体代码如下:

# coding: utf-8
import tensorflow as tf 
import numpy as np 

# 创造数据,y = 0.1x + 0.3
x_data = np.random.rand(100).astype(np.float32)
y_data = x_data * 0.1 + 0.3

### 开始创建 tensorflow ###

# 可能为矩阵,首字母大写
# 随机数列生成,一维的,范围[-1, 1]的数
Weights = tf.Variable(tf.random_uniform([1], -1.0, 1.0))
# 初始值为 0 的一维的数
biases = tf.Variable(tf.zeros([1]))

y = Weights * x_data + biases

# 预测值与真实值的差别
loss = tf.reduce_mean(tf.square(y - y_data))

# 建立优化器,减少误差,提高参数的准确度
# 学习效率是一个小于 1 的数,这里设为 0.5
optimizer = tf.train.GradientDescentOptimizer(0.5)
train = optimizer.minimize(loss)

# 初始化
init = tf.global_variables_initializer()

### 创建完毕 ###

# 激活 tensorflow
sess = tf.Session()
sess.run(init)

for step in range(201):
    sess.run(train)
    # 每隔 20 步输出运行结果
    if step % 20 == 0:
        print step, sess.run(Weights), sess.run(biases)


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值