首页 > 其他 > 详细

tensorboard

时间:2018-10-18 21:31:50      阅读:24      评论:0      收藏:0      [点我收藏+]

标签:variable   rain   sum   需要   oba   fault   color   loss   range   

import tensorflow as tf
import numpy as np
from matplotlib import pyplot as plt

# 清除
tf.reset_default_graph()
logdir = log
plt.figure()

x_data = np.random.rand(1000).astype(np.float32)
y_data = x_data * 8 + 1.5

plt.plot(x_data, y_data)
plt.show()
x = tf.placeholder(tf.float32)
y = tf.placeholder(tf.float32)

w = tf.Variable(tf.random_normal([1]), tf.float32)
b = tf.Variable(tf.random_normal([1]), tf.float32)

y_pre = tf.multiply(x, w) + b

loss = tf.reduce_sum(tf.pow((y - y_pre), 2))
train = tf.train.GradientDescentOptimizer(0.01).minimize(loss)

init = tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init)
    for i in range(1000):
        sess.run(train, feed_dict={x: x_data[i], y: y_data[i]})
        if i % 50 == 0:
            print(sess.run(loss, feed_dict={x: x_data[i], y: y_data[i]}))

    writer = tf.summary.FileWriter(logdir, tf.get_default_graph())
    writer.close()

1. tf.reset_default_graph() 是清除default gragh 和不断增加的节点

2.定义一个writer,参数为log的地址,和图形这里我们直接用get_default_graph()来获得tensorflow默认生成的图

3.需要把write 关闭了

tensorboard

标签:variable   rain   sum   需要   oba   fault   color   loss   range   

原文:https://www.cnblogs.com/francischeng/p/9813037.html

(0)
(0)
   
举报
评论 一句话评论(0
0条  
登录后才能评论!
© 2014 bubuko.com 版权所有 鲁ICP备09046678号-4
打开技术之扣,分享程序人生!
             

鲁公网安备 37021202000002号