☰

Keras の Tensorboard コールバックを使用するにはどうすればいいですか? 質問する

tag-icon • tag-icon keras tensorboard
Keras の Tensorboard コールバックを使用するにはどうすればいいですか? 質問する

私は Keras を使ってニューラル ネットワークを構築しました。そのデータを Tensorboard で視覚化するため、次のものを利用しました。

keras.callbacks.TensorBoard(log_dir='/Graph', histogram_freq=0,
                            write_graph=True, write_images=True)

で説明したようにケラスコールバックを実行すると が返されます<keras.callbacks.TensorBoard at 0x7f9abb3898>が、フォルダー「Graph」にファイルが何も表示されません。このコールバックの使用方法に何か問題があるのでしょうか?

ベストアンサー1

keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=0,  
          write_graph=True, write_images=True)

この行は Callback Tensorboard オブジェクトを作成するので、そのオブジェクトをキャプチャしてfitモデルの関数に渡す必要があります。

tbCallBack = keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=0, write_graph=True, write_images=True)
...
model.fit(...inputs and parameters..., callbacks=[tbCallBack])

この方法では、コールバック オブジェクトを関数に渡します。これはトレーニング中に実行され、Tensorboard で使用できるファイルを出力します。

トレーニング中に作成されたファイルを視覚化したい場合は、ターミナルで実行します。

tensorboard --logdir path_to_current_dir/Graph

おすすめ記事