如何解决是否有一个函数可以总结类似于 Keras 的 model.summary() 的 tf.estimator.Estimator 模型?
我想获得 tf.estimator.Estimator
中使用的 Nural Network 模型的摘要信息。是否有任何函数可以用来显示类似于 keras model.summary()
显示的内容?例如
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
dense_16 (Dense) (None,3) 21
_________________________________________________________________
dense_17 (Dense) (None,3) 12
_________________________________________________________________
dense_18 (Dense) (None,1) 4
=================================================================
Total params: 37
Trainable params: 37
Non-trainable params: 0
_________________________________________________________________
None
我尝试使用model.get_variable_names()
和model.get_variable_values(param_name)
,但是很难知道图层信息
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。