2017-06-13 168 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

相关问题