码迷,mamicode.com
首页 > 其他好文 > 详细

Tensorflow Practice 2-4

时间:2018-12-18 02:07:48      阅读:241      评论:0      收藏:0      [点我收藏+]

标签:tensor   tde   ini   nump   rac   优化器   step   a*   class   

 1 import tensorflow as tf
 2 import numpy as np
 3 
 4 ‘‘‘使用numpy生成100个随机点‘‘‘
 5 x_data = np.random.rand(100)
 6 y_data = x_data*0.1 + 0.2
 7 
 8 ‘‘‘构造一个线性模型‘‘‘
 9 b = tf.Variable(0.)
10 k = tf.Variable(0.)
11 y = k*x_data + b
12 
13 ‘‘‘二次代价函数‘‘‘
14 loss = tf.reduce_mean(tf.square(y_data - y))
15 ‘‘‘定义一个梯度下降法来进行训练的优化器‘‘‘
16 optimizer = tf.train.GradientDescentOptimizer(0.2)
17 ‘‘‘最小化代价函数‘‘‘
18 train = optimizer.minimize(loss)
19 
20 ‘‘‘初始化变量‘‘‘
21 init = tf.global_variables_initializer()
22 
23 with tf.Session() as sess:
24     sess.run(init)
25     for step in range(201):
26         sess.run(train)
27         if step % 20 == 0:
28             print(step, sess.run([k, b]))

 

Tensorflow Practice 2-4

标签:tensor   tde   ini   nump   rac   优化器   step   a*   class   

原文地址:https://www.cnblogs.com/StevenSun1991/p/10134746.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!