我有一个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上可视化在训练期间损失随着时期而减少。实现这一目标的最佳方法是什么?