[学习方法]深度学习入门系列--tensorflow入门

      这是一个非常简单的线性回归的例子,把例子的实现步骤搞清楚,可以实现对tensorflow训练框架的基本了解。

import tensorflow as tf 
import numpy as np
import matplotlib.pyplot as plt
#第一步:定义超参数
leraining_rate=0.01
global_step=20

#第二步:处理数据
x = np.linspace(-2, 2, 100)
y = 3 * x + np.random.randn(*x.shape) * 0.4 +4

#第三步:定义模型(线性回归)
X=tf.placeholder(tf.float32,name="input")
w=tf.Variable(0.0)
b=tf.Variable(0.0)
Y=X*w+b

#第四步:定义损失函数
Y_=tf.placeholder(tf.float32,name="output")
loss = tf.square(Y - Y_)

#第五步:定义优化器
optimizer=tf.train.GradientDescentOptimizer(leraining_rate)

#第六步:定义单步训练
train_op=optimizer.minimize(loss)


#第七步:定义会话,开始循环训练
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
	#print("start train:")
    for i in range(global_step):
        for (xi, yi) in zip(x, y):
            sess.run(train_op,feed_dict={X:xi,Y_:yi})
            step_loss=sess.run(loss,feed_dict={X:xi,Y_:yi})
        print(step_loss)
        weight=sess.run(w,feed_dict={X:xi,Y_:yi})
        bias=sess.run(b,feed_dict={X:xi,Y_:yi})
        print("weight=%f,bias=%s"%(weight,bias))

#第八步:结果可视化
plt.plot(x,y,"*")
plt.plot(x,x*weight+bias)
plt.show()
	

结果可视化

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值