说明:
本文是根据优酷 莫凡 python tensorflow 中第14节视频
使用的环境为:
Ubuntu 14,04LST x86_64
GPU : nivdia K80
好,我们开始吧。
这次我们会介绍如何可视化神经网络。因为很多时候我们都是做好了一个神经网络,但是没有一个图像可以展示给大家看。这一节会介绍一个TensorFlow的可视化工具—tensorboard :)
通过使用这个工具我们可以很直观的看到整个神经网络的结构、框架。
以前几节的代码为例:相关代码
好,通过阅读代码我们大概知道了此处是有一个输入层(inputs),一个隐含层(layer),还有一个输出层(output)
现在可以看看如何进行可视化
首先从Input 开始:
# define placeholder for inputs to network
xs = tf.placeholder(tf.float32, [None, 1])
ys = tf.placeholder(tf.float32, [None, 1])
对于input我们进行如下修改:
首先,可以为xs指定名称为x_in:
xs= tf.placeholder(tf.float32, [None, 1],name='x_in')
然后再次对ys 指定 名称y_in
ys= tf.placeholder(tf.loat32, [None, 1],name='y_in')
这里指定的名称将来会在可视化的图层inputs中显示出来
使用with tf.name_scope(‘inputs’)可以将xs和ys包含进来,形成一个大的图层,图层的名字就是with tf.name_scope()方法里的参数。
with tf.name_scope('inputs'):
# define placeholder for inputs to network
xs = tf.placeholder(tf.float32, [None, 1])
ys = tf.placeholder(tf.float32, [None, 1])
接下来开始编辑layer , 请看未编辑前的程序片段 :
def add_layer(inputs, in_size, out_size, activation_function=None):
# add one more layer and return the output of this layer
Weights = tf.Variable(tf.random_normal([in_size, out_size]))
biases = tf.Variable(tf.zeros([1, out_size]) + 0.1)
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
这里的名字应该叫layer ,
def add_layer(inputs, in_size, out_size, activation_function=None):
# add one more layer and return the output of this layer
with tf.name_scop('layer'):
Weights= tf.Variable(tf.random_normal([in_size, out_size]))
## and so on...
在定义完大的框架layer之后,同时也需要定义每一个’框架‘里面的小部件:(Weights biase和activation function):
现在现对Weights 定义:
定义的方法同上,可以使用tf.name.scop()方法,同时也可以在Weights中指定名称W。
即为:
def add_layer(inputs, in_size, out_size, activation_function=None):
#define layer name
with tf.name_scop('layer'):
##define weights name
with tf.name_scop('weights')
Weights= tf.Variable(tf.random_normal([in_size, out_size]),name='W')
###and so on......
接着继续定义biase , 定义方式同上。
def add_layer(inputs, in_size, out_size, activation_function=None):
#define layer name
with tf.name_scop('layer'):
##define weights name
with tf.name_scop('weights')
Weights= tf.Variable(tf.random_normal([in_size, out_size]),name='W')
## define biase
with tf.name_scope('Wx_plus_b'):
Wx_plus_b = tf.add(tf.matmul(inputs, Weights), biases)
### and so on....
activtion_function 的话,可以暂时忽略。因为当你自己选择用tensorflow中的激励函数(activation function)的时候,tensorflow会默认添加名称。
最终,layer形式如下:
def add_layer(inputs, in_size, out_size, activation_function=None):
# add one more layer and return the output of this layer
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
最后编辑loss部分:将with tf.name_scope()添加在loss上方,并为它起名为loss
# the error between prediciton and real data
with tf.name_scope('loss'):
loss = tf.reduce_mean(tf.reduce_sum(tf.square(ys - prediction),eduction_indices=[1]))
使用with tf.name_scop()再次对train_step部分进行编辑,如下:
with tf.name_scope('train'):
train_step = tf.train.GradientDescentOptimizer(0.1).minimize(loss)
我们需要使用tf.train.SummaryWriter()将上面‘绘画’出的图保存到一个目录中,以方便后期在浏览器中可以浏览。这个方法中的第二个参数需要使用sess.graph , 因此我们需要把这句话放在获取session的后面。这里的graph是将前面定义的框架信息收集起来,然后放在logs/目录下面。
sess = tf.Session() ##get session
writer = tf.train.SummaryWriter("logs/", sess.graph)
最后在你的terminal(终端)中 ,使用以下命令
tensorborad --logdir='logs/'
同时将终端中输出的网址复制到浏览器中, 便可以看到之前定义的视图框架了。
tensorboard 还有很多其他的参数,希望大家可以多多了解、
最终的全部代码在这里