Tensorflow深度学习笔记(三)-TensorFlow基本应用(梯度下降法)

来源:互联网 发布:js tbody 动态添加行 编辑:程序博客网 时间:2024/04/29 00:29

本文主要引入一个简单的示例来阐述Tensorflow的应用。

先来一段程序,它是采用梯度下降法来训练数据。

import  tensorflow as tfimport numpy as np #numpy库为数学计算库,若导入失败,请先安装(pip install numpy)#使用numpy随机产生100个随机点x_data=np.random.rand(100)y_data = x_data+1#构造一个线性模型b= tf.Variable(0.)k= tf.Variable(0.)y= k*x_data +b#损失函数loss = tf.reduce_mean(tf.square(y_data-y))#定义一个梯度下降法优化器optimizer = tf.train.GradientDescentOptimizer(0.03)#最小化代价函数train = optimizer.minimize(loss)#初始化全局变量init = tf.global_variables_initializer()with tf.Session() as sess:    sess.run(init)    #训练2000次    for step in range(2001):        sess.run(train)        #每训练20次打印输出训练次数和得到的k,b值        if step%20 == 0:            print(step,sess.run([k,b]))

本文主要作为一个引入,还有一些概念不懂的,可以自行去查询相关内容,后面会继续分析相关内容。

Enjoy.


转载请注明出处,Juyin@2017/11/12

阅读全文
0 0