らんだむな記憶

blogというものを体験してみようか!的なー

Tensorboardで可視化

とか、情報を収集しつつ。
Udacityのud187の課題を使う。

from __future__ import absolute_import, division, print_function, unicode_literals
import os
import numpy as np
import tensorflow as tf

# TF 1.14.0くらい向け。
tf.enable_eager_execution()

def learn():
    celsius_q    = np.array([-40, -10,  0,  8, 15, 22,  38],  dtype=float)
    fahrenheit_a = np.array([-40,  14, 32, 46, 59, 72, 100],  dtype=float)

    # モデルの構築
    model = tf.keras.Sequential([
        tf.keras.layers.Dense(units=1, input_shape=[1])
    ])

    # モデルのコンパイル
    model.compile(loss='mean_squared_error',
                  optimizer=tf.keras.optimizers.Adam(0.1))

    # TensorBoardのセットアップ
    log_filepath = os.path.join(root_dir, "logs")
    tb_cb = tf.keras.callbacks.TensorBoard(log_dir=log_filepath, 
        histogram_freq=1, write_graph=True, write_images=True)
    cbks = [tb_cb]

    history = model.fit(celsius_q, fahrenheit_a, epochs=500, verbose=False, callbacks=cbks)

これでログが出たので、

tensorboard --logdir=./logs

してみる。よく見るようなCNNのニューロンが繋がった図が出るわけではないけど、まぁなんか可視化できるんだなぁと。
f:id:derwind:20190916004319p:plain