欢迎使用CSDN-markdown编辑器

来源:互联网 发布:一件代发怎么传淘宝 编辑:程序博客网 时间:2024/06/03 21:13

tensorflow 中文文档学习

import tensorflow as tfimport numpy as npx_data = np.float32(np.random.rand(2,100))y_data = np.dot([0.1,0.2],x_data)+0.3# construct liner_model   这里写中文会出错的(注释也不行)  如果写中文 加上utf-8# 构建线性模型 b = tf.Variable(tf.zeros([1]))  # tf.zeros(shape)W = tf.Variable(tf.random_uniform([1,2],-1,1))y = tf.matmul(W,x_data)+b#定义损失函数loss = tf.reduce_mean(tf.square(y-y_data))#定义优化函数optimizer = tf.train.GradientDescentOptimizer(0.5)#定义训练optrain = optimizer.minimize(loss)# 初始化  一定要初始化!!!init = tf.initialize_all_variables()sess = tf.Session()sess.run(init)for step in xrange(0,201):    sess.run(train)    if step%20 == 0:        print step,sess.run(W),sess.run(b)sess.close()
原创粉丝点击