TensorBoard菜鸟教程(包含TFlearn例子)

最新推荐文章于 2025-05-07 14:50:32 发布

英国老鼠_

最新推荐文章于 2025-05-07 14:50:32 发布

阅读量3.5k

点赞数 3

分类专栏： Tensorflow 深度学习

本文链接：https://blog.csdn.net/hwj_wayne/article/details/78224599

版权

这篇博客针对TensorBoard的使用提供了详细的新手教程，包括如何启动TensorBoard，通过实例解析计算图的结构，以及如何在TFlearn中生成events文件。通过示例代码，解释了如何利用TensorBoard进行模型可视化和数据跟踪。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 简介

网上关于TensorBoard有很多介绍，但作为一名小白很难操作起来，实现过程中困难重重。本文章从实例解析tensorboard的使用方法。其他文字方面的介绍（如TensorBoard是什么、TensorBoard的作用）可参考大神们的博客。以下代码转自http://www.jianshu.com/p/61081bba175f 。已运行通过（python3）

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("/tmp/MNIST_data", one_hot=True)

# Input placeholder, 2-D tensor of floating-point nunbers.
# here None means that a dimension can be of any length.
X = tf.placeholder(tf.float32, [None, 784], name = 'X-input')

# New placeholder to input the correct answers.
Y = tf.placeholder(tf.float32, [None, 10], name = 'Y-input')

# Initialize both W and b as tensors full of zeros.
# Since we are going to learn W and b, it doesn't matter very much what they initial are.
W = tf.Variable(tf.zeros([784, 10]), name = 'Weight')
B = tf.Variable(tf.zeros([10]), name = 'Bias')

# Tensorboard histogram summary.
tf.summary.histogram('WeightSM', W)
tf.summary.histogram('BiasSM', B)

with tf.name_scope('Layer'):
    y = tf.nn.softmax(tf.matmul(X, W) + B)

with tf.name_scope('Cost'):
    cross_entropy = tf.reduce_mean(-tf.reduce_sum(Y * tf.log(y), reduction_indices=[1]))
    # Tensorboard scalar summary.
    tf.summary.scalar('Cost', cross_entropy)

with tf.name_scope('Train'):
    train_step = tf.train.GradientDescentOptimizer(0.5).minimize(cross_entropy)

with tf.name_scope('Accuracy'):
    accuracy = tf.reduce_mean(tf.cast(tf.equal(tf.argmax(y, 1), tf.argmax(Y, 1)), tf.float32))
    # Tensorboard scalar summary.
    tf.summary.scalar('Accuracy', accuracy)

with tf.Session() as sess:
    # Merge all summaries.
    writer=tf.summary.FileWriter('./logs',sess.graph)
    merged = tf.summary.me

最低0.47元/天解锁文章