[Tensorflow] Cookbook - Tensorboard***

Ref: https://www.tensorflow.org/get_started/summaries_and_tensorboard

 

可视化对于Training的重要性，不言而喻。

---

 

• 代码示范

# -*- coding: utf-8 -*-
# Using Tensorboard
#----------------------------------
#
# We illustrate the various ways to use
#  Tensorboard

import os
import io
import time
import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf

# Initialize a graph session
sess = tf.Session()

# Create a visualizer object
summary_writer = tf.train.SummaryWriter('tensorboard', tf.get_default_graph())

# Create tensorboard folder if not exists
if not os.path.exists('tensorboard'):
    os.makedirs('tensorboard')
print('Running a slowed down linear regression. '
      'Run the command: $tensorboard --logdir="tensorboard"  '
      ' Then navigate to http://127.0.0.0:6006')

# You can also specify a port option with --port 6006

# Wait a few seconds for user to run tensorboard commands
time.sleep(3)

# Some parameters
batch_size  = 50
generations = 100

# Create sample input data
x_data     = np.arange(1000)/10.
true_slope = 2.
y_data     = x_data * true_slope + np.random.normal(loc=0.0, scale=25, size=1000)

【构造好了ground true数据】

# Split into train/test
train_ix = np.random.choice(len(x_data), size=int(len(x_data)*0.9), replace=False)
test_ix  = np.setdiff1d(np.arange(1000), train_ix)　　# 提取出setdiff1d不同的部分(only index)
x_data_train, y_data_train = x_data[train_ix], y_data[train_ix]
x_data_test,  y_data_test  = x_data[test_ix ], y_data[test_ix ]

# Declare placeholders 加载样本的容器
x_graph_input = tf.placeholder(tf.float32, [None])
y_graph_input = tf.placeholder(tf.float32, [None])

# Declare model variables
m = tf.Variable(tf.random_normal([1], dtype=tf.float32), name='Slope')

# Declare model: Input layer + weight --> value of next layer
output = tf.mul(m, x_graph_input, name='Batch_Multiplication')

# Declare loss function (L1)
residuals = output - y_graph_input　　# 联想到了 "深度残差网络" 何凯明，减小均值
l2_loss   = tf.reduce_mean(tf.abs(residuals), name="L2_Loss")

# Declare optimization function
my_optim = tf.train.GradientDescentOptimizer(0.01)　　# 通过这个solver缩小loss
train_step = my_optim.minimize(l2_loss)

# Visualize a scalar
with tf.name_scope('Slope_Estimate'):
    tf.scalar_summary('Slope_Estimate', tf.squeeze(m))
    
# Visualize a histogram (errors)
with tf.name_scope('Loss_and_Residuals'):
    tf.histogram_summary('Histogram_Errors', l2_loss)
    tf.histogram_summary('Histogram_Residuals', residuals)


# Declare summary merging operation
summary_op = tf.merge_all_summaries()

【op操作各种各样，所以需要有个汇总的op操作】

# Initialize Variables
init = tf.initialize_all_variables()
sess.run(init)

for i in range(generations):
    batch_indices = np.random.choice(len(x_data_train), size=batch_size)
    x_batch = x_data_train[batch_indices]
    y_batch = y_data_train[batch_indices]
    _, train_loss, summary = sess.run([train_step, l2_loss, summary_op],
                                      feed_dict={x_graph_input: x_batch, y_graph_input: y_batch})
    
    test_loss, test_resids   = sess.run([l2_loss, residuals], feed_dict={x_graph_input: x_data_test, y_graph_input: y_data_test})
    
    if (i+1)%10==0:
        print('Generation {} of {}. Train Loss: {:.3}, Test Loss: {:.3}.'.format(i+1, generations, train_loss, test_loss))

    log_writer = tf.train.SummaryWriter('tensorboard')
    log_writer.add_summary(summary, i)
    time.sleep(0.5)


#Create a function to save a protobuf bytes version of the graph
def gen_linear_plot(slope):
    linear_prediction = x_data * slope
    plt.plot(x_data, y_data, 'b.', label='data')
    plt.plot(x_data, linear_prediction, 'r-', linewidth=3, label='predicted line')
    plt.legend(loc='upper left')
    buf = io.BytesIO()
    plt.savefig(buf, format='png')
    buf.seek(0)
    return(buf)

# Add image to tensorboard (plot the linear fit!)
slope = sess.run(m)
plot_buf = gen_linear_plot(slope[0])
# Convert PNG buffer to TF image
image = tf.image.decode_png(plot_buf.getvalue(), channels=4)
# Add the batch dimension
image = tf.expand_dims(image, 0)
# Add image summary
image_summary_op = tf.image_summary("Linear Plot", image)
image_summary    = sess.run(image_summary_op)
log_writer.add_summary(image_summary, i)
log_writer.close()

 

• 查看网络结构

 

• 实时跟踪权重

Ref: http://www.jianshu.com/p/52e773d47338

tensorboard --logdir results --reload_interval 5 
【默认的 reload_interval 是120秒，以避免在计算机上面太快统计，但是在我们的情况下，我们可以安全地加速一点】

 

 

06：Tensorflow的可视化工具Tensorboard的初步使用

小姑娘整理的不错，之后二次整理一下。