如何通过keras在tensorboard中的同一图表上显示训练损失和验证损失?

时间:2017-06-13 07:48:36

标签: keras tensorboard

我使用带有Tensorflow后端的Keras训练CNN,我使用张量板来显示损失函数和准确度。我想在同一个图表上看到训练数据和验证数据的损失函数,但是我在使用Tensorflow而不是通过keras时只找到了这样做的方法。

有办法吗?

编辑1: 我尝试在Regex中编写loss / 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))

2 个答案:

答案 0 :(得分: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)

我是从Github上找到这个目的的,但没有使用tensorboard。希望这会有所帮助!

live loss plot for keras