Jeg har bygget et nevralt nettverk med Keras. Jeg ville visualisere dataene ved hjelp av Tensorboard, derfor har jeg brukt:
keras.callbacks.TensorBoard(log_dir='/Graph', histogram_freq=0,
write_graph=True, write_images=True)
som forklart i keras.io. Når jeg kjører callbacken får jeg <keras.callbacks.TensorBoard at 0x7f9abb3898>
, men jeg får ikke opp noen fil i mappen "Graph". Er det noe galt i hvordan jeg har brukt denne tilbakeringingen?
keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=0,
write_graph=True, write_images=True)
Denne linjen oppretter et Callback Tensorboard-objekt, du bør fange opp dette objektet og gi det til fit
-funksjonen i modellen din.
tbCallBack = keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=0, write_graph=True, write_images=True)
...
model.fit(...inputs and parameters..., callbacks=[tbCallBack])
På denne måten gir du callback-objektet til funksjonen. Den kjøres under treningen og sender ut filer som kan brukes med tensorboard.
Hvis du vil visualisere filene som opprettes under treningen, kan du kjøre følgende i terminalen din
tensorboard --logdir path_to_current_dir/Graph
Håper dette er til hjelp!
Endring
keras.callbacks.TensorBoard(log_dir='/Graph', histogram_freq=0,
write_graph=True, write_images=True)
til
tbCallBack = keras.callbacks.TensorBoard(log_dir='Graph', histogram_freq=0,
write_graph=True, write_images=True)
og angi modellen din
tbCallback.set_model(model)
Kjør i terminalen
tensorboard --logdir Graph/