在Keras中显示Tensorflow模型摘要

时间:2018-04-16 19:25:37

标签: tensorflow machine-learning keras

我们可以使用tensorflow图层构建模型。有没有什么方法可以像Keras一样显示模型摘要。

Keras Model Summary

2 个答案:

答案 0 :(得分:1)

不,没有这样的选择。 TensorFlow比Keras更通用,允许任意图形架构,因此显示这样的结构化摘要对任意TensorFlow图形没有意义。最接近的可能是TensorBoard,它有一个非常方便的交互式graph visualization tool

答案 1 :(得分:0)

Keras是TensorFlow的一部分(有一段时间),因此您总是可以得到类似以下的好东西:

model.output_shape # model summary representation
model.summary() # model configuration
model.get_config() # list all weight tensors in the model
model.get_weights() # get weights and biases