ColabolaboratoryでTensorBoardを使う

TensorBoard extensionの読み込みなど

%load_ext tensorboard
import datetime

TensorBoardで使用するデータは./logs/に格納されます
logsディレクトリ不要なデータが無いように一度ディレクトリを削除します

!rm -rf ./logs/ 

Kerasにはcallbackと呼ばれるものが用意されており，Tensorboard用のlogを出力するときには、この仕組みを使用する。Callbackは決まったタイミングで呼ばれ、そのタイミングは、Tensorboard用のCallbackの場合は、学習の直前と毎epochの終了時に呼び出されlogが保存されます。

次の例ではlogファイルは./logs/fit/ディレクトリに作成(log_dir)

log_dir="logs/fit/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)

作成したcallbackは、Model.fit()関数の中に登録。これによりlogs/fit/ディレクトリ内にログが保存される

model.fit(x=x_train, 
          y=y_train, 
          epochs=5, 
          validation_data=(x_test, y_test), 
          callbacks=[tensorboard_callback]) #作成したcallbackの指定

TensorBoardの起動

%tensorboard --logdir logs/fit