在Keras Functional API

时间:2016-10-12 20:53:17

标签: callback keras tensorboard

我正在https://github.com/fchollet/keras/blob/master/examples/pretrained_word_embeddings.py

的预训练单词嵌入模型示例中学习

我想使用Tensorboard可视化培训过程。 Keras在https://keras.io/callbacks/#tensorboard处有Sequential API的Tensorboard回调,但该示例使用的Model API没有。

如何使用张量板可视化Model API网络?

1 个答案:

答案 0 :(得分:1)

回调对两种API同样有效 我目前在使用Model API构建的模型中使用Tensorboard回调,因此我可以确认 在使用任何API定义模型并简单地编译它call

之后
tensorboard = keras.callbacks.TensorBoard()

model.fit(self, x, y, callbacks=[tensorboard])