tensorflow shizhan

import tensorflow as tf
import numpy as np
x=np.random.rand(100).astype("float32")
y=x*0.1+0.3
w=tf.Variable(tf.random_uniform([1],-1.0,1.0))
b=tf.Variable(tf.zeros([1]))
yy=w*x+b
loss=tf.reduce_mean(tf.square(yy-y))
optimizer=tf.train.GradientDescentOptimizer(0.5)
train=optimizer.minimize(loss)
init=tf.initialize_all_variables()
sess=tf.Session()
sess.run(init)
for step in xrange(201):
    sess.run(train)
    if step%20==0:
        print(step,sess.run(w),sess.run(b))
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章