Как получить гистограмму с тензорной доски в Google Colab - PullRequest
Новая
       14

Как получить гистограмму с тензорной доски в Google Colab

0 голосов
/ 25 ноября 2018

Я использую пакет под названием 'tensboardboardabab'.Но сейчас я могу только получить график из тензорной доски на основе следующего кода: 

!pip install -U tensorboardcolab
from tensorboardcolab import *

tbc=TensorBoardColab()

tf.reset_default_graph()

with tf.get_default_graph().as_default():
    x = tf.placeholder(tf.float32, [], name="x")
    y = tf.placeholder(tf.float32, [], name="y")
    k = tf.get_variable("k", shape=[])
    b = tf.get_variable("b", shape=[])
    loss = tf.square(k * x + b - y, name="square_error")

optimizer = tf.train.GradientDescentOptimizer(0.5)
train_op = optimizer.minimize(loss)

with tf.Session() as sess:
# Initialize `k` and `b`
sess.run(tf.global_variables_initializer())
for i in range(100):
    # y = 3x-1 + noise
    data_x = np.random.random()
    data_y = 3.0 * data_x - 1.0 + 0.0001*np.random.randn()

    k_val, b_val, loss_val, _ = sess.run([k, b, loss, train_op],
                                         feed_dict={x: data_x, y: data_y})
    if i % 20 == 0:
        print("k, b, loss: ", k_val, b_val, loss_val)

sess.close()

train_writer = tbc.get_writer();
train_writer.add_graph(sess.graph)

train_writer.flush();
tbc.close()

Могу ли я добавить некоторый код для визуализации гистограммы ??Я не имел ни малейшего понятия, чтобы найти, чтобы заполнить параметры, такие как «write_grads = True» и т. Д.

...