1. ホーム
  2. keras

[解決済み] KerasのTensorboardコールバックはどのように使うのですか?

2022-04-27 21:36:25

質問

Kerasでニューラルネットワークを構築しました。そのデータをTensorboardで可視化したいと思い、利用させていただいています。

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

で説明したように keras.io . コールバックを実行すると、次のようになります。 <keras.callbacks.TensorBoard at 0x7f9abb3898> しかし、私のフォルダ "Graph"には、何のファイルもありません。このコールバックの使い方に何か間違いがあるのでしょうか?

解決方法を教えてください。

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 

これが役に立つといいのですが......。