如何在keras模型受到训练的同时在张量板上看到损失与时代图?

时间:2018-05-25 05:54:53

标签: python tensorflow keras tensorboard loss-function

我有一个2层keras顺序模型,带有2个Dense图层,用于执行回归。 我的顺序模型如下:

  model = Sequential()
  model.add(Dense(7, input_dim=7))
  model.add(Dense(1))
  model.compile(loss='mse',optimizer='rmsprop')

我初始化了tensorboard(keras.callbacks.Tensorboard)并将其作为回调传递给了model.fit方法。

tensorboard = TensorBoard(log_dir='logs/', histogram_freq=1, write_graph=True, write_images=True)
model.fit(X_train, y_train, callbacks=[tensorboard])

在启动张量板时,我可以看到计算图,但我也希望在Tensorboard上可视化在训练期间损失随着时期而减少。实现这一目标的最佳方法是什么?

0 个答案:

没有答案