from __future__ import print_function , unicode_literals , absolute_import , division
import tensorflow as tf
定义添加层的函数
def add_layer(inputs , in_size , out_size , activation_function=None ):
with tf.name_scope('layer'):
with tf.name_scope('weights'):
Weights = tf.Variable(tf.random_normal([in_size , out_size]) , name='W')
with tf.name_scope('biases'):
biases = tf.Variable(tf.zeros([1 , out_size]) + 0.1 , name='b' )
with tf.name_scope('Wx_plus_b'):
Wx_plus_b = tf.add(tf.matmul( inputs , Weights ) , biases)
if activation_function is None :
outputs = Wx_plus_b
else:
outputs = activation_function( Wx_plus_b )
return outputs
定义xs,ys用于输入数据
with tf.name_scope('inputs'):
xs = tf.placeholder( tf.float32 , [ None , 1 ] , name='x_input')
ys = tf.placeholder( tf.float32 , [ None , 1 ] , name='y_input')
添加层
l1 = add_layer( xs , 1 , 10 , activation_function=tf.nn.relu )
prediction = add_layer( l1 , 10 , 1 , activation_function=None )
损失率
with tf.name_scope('loss'):
loss = tf.reduce_mean(tf.reduce_sum(tf.square(ys - prediction) , reduction_indices=[1]))
训练方式
with tf.name_scope('train'):
train_step = tf.train.GradientDescentOptimizer(0.1).minimize(loss)
建立会话
sess = tf.Session()
writer = tf.summary.FileWriter("logs/" , sess.graph)
- 此处运行完之后会生成一个文件位于logs目录下;
- 进入cmd命令行模式切换到正确的路径下,我的logs路径是:C:\Users\24301(\logs),括号内的不需要,cd
C:\Users\24301; - 激活tensorflow
- 输入tensorboard --logdir=logs,复制生成的网址路径,在浏览器中打开
初始化
sess.run(tf.global_variables_initializer())