Comment utiliser le rappel Tensorboard de Keras?

Question

J'ai construit un réseau de neurones avec Keras. Je voudrais visualiser ses données par Tensorboard, donc j'ai utilisé:

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

comme expliqué dans keras.io . Lorsque je lance le rappel, j'obtiens <keras.callbacks.TensorBoard at 0x7f9abb3898> , mais je n'ai aucun fichier dans mon dossier "Graph". Quelque chose ne va pas dans la façon dont j'ai utilisé ce rappel?

Answer 1

Voici comment vous utilisez le rappel TensorBoard :

 from keras.callbacks import TensorBoard

tensorboard = TensorBoard(log_dir='./logs', histogram_freq=0,
                          write_graph=True, write_images=False)
# define model
model.fit(X_train, Y_train,
          batch_size=batch_size,
          epochs=nb_epoch,
          validation_data=(X_test, Y_test),
          shuffle=True,
          callbacks=[tensorboard])
 

Answer 2

Changement

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

à

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

et définir votre modèle

 tbCallback.set_model(model)
 

Courez dans votre terminal

 tensorboard  --logdir Graph/