2017-06-13 170 views
3

我使用Keras和Tensorflow後端來訓練CNN,我使用張量板來顯示損失函數和準確性。我希望在同一個圖表上看到訓練數據和驗證數據的損失函數,但我只在使用Tensorflow而不是通過keras時找到了方法。如何通過keras在tensorboard中的同一圖表上顯示培訓損失和驗證損失?

有沒有辦法做到這一點?

編輯1: 我試着寫損失/ ACC的正則表達式,但不是把兩個圖表在一起就說明他們像這樣並排: http://imgur.com/a/oLIcL

伊夫加入我用來登錄到張量板:

tbCallBack=keras.callbacks.TensorBoard(log_dir='C:\\logs', histogram_freq=0, write_graph=False, write_images=True, embeddings_freq=0, embeddings_layer_names=None, embeddings_metadata=None) 

model.fit_generator(train_generator, 
       steps_per_epoch=x_train.shape[0] // batch_size, 
       epochs=epochs, 
validation_data=(x_test, y_test)) 

回答

0

您可以在Tensorboard窗口左上角的文本框中添加正則表達式。

爲列車/​​驗證數據的準確性添加acc。爲損失值添加loss。這對Keras和Tensorflow都適用。

從這個漂亮的教程TB得到這個:https://www.youtube.com/watch?v=eBbEDRsCmv4

作爲一個代碼片段我用這個:

logdir = "_tf_logs/" + now.strftime("%Y%m%d-%H%M%S") + "/" 
tb = TensorBoard(log_dir=logdir) 
callbacks=[tb] 
... 
model.fit(X_train, Y_train, validation_data=val_data, epochs=10, verbose=2, callbacks=callbacks) 
+1

謝謝回答!我已經解決了您在編輯中編寫的內容,因爲它更容易通過此處。 也是你分享的鏈接鏈接到谷歌搜索,而不是你試圖鏈接的教程。 –

+0

謝謝。鏈接已更正。也非常感謝你指出了這一點。 – petezurich

相關問題