tensorflow v.2.x で tensorboard in keras

• 2020/08/13

参考

• TensorBoardスカラー：Kerasでのトレーニング指標のロギング
  • tensorflow 公式
  • 「カスタムスカラーのロギング」を参考
  • file_writer を作成して、tf.summary.xxx で書き込む

# writer を作成

file_writer = tf.summary.create_file_writer(logdir + "/metrics")

# 書き込む

tf.summary.scalar('learning rate', data=learning_rate, step=epoch)

• TensorBoardでの画像データの表示
  • tensorflow 公式
  • training は fit function を使いたかったので、「画像分類器の作成」が参考になった
  • TensorBoard class と writer (tf.summary.create_file_writer)をどちらも作成
  • epoch ごと(もしくは batchごと)に実行したい処理を関数で作成
  • keras.callbacks.LambdaCallback で使用できる形に
  • fit 実行時に callback に含める

tensorboard_callback = keras.callbacks.TensorBoard(log_dir=logdir)

file_writer_cm = tf.summary.create_file_writer(logdir + '/cm')

# 実行する処理

cm_callback = keras.callbacks.LambdaCallback(on_epoch_end=log_confusion_matrix)

# callbackに含める

model.fit(

    train_images,

    train_labels,

    epochs=5,

    verbose=0, # Suppress chatty output

    callbacks=[tensorboard_callback, cm_callback],

    validation_data=(test_images, test_labels),

)