机器学习模型的可视化

tensorflow veyvin 2612℃ 0评论
文章目录[隐藏]
本文最后更新于2020年8月2日 13:34,如果文章内容失效,请反馈给我们,谢谢!

机器学习研究有一段时间了,期间研究了tensorflow,keras和cntk.

神经网络的只有在实验的情况下才能知道这个AI神经网络的好坏.

前人的研究形成了一系列准确率非常高的神经网络.

tensorflow自带读取网络工具

通过tensorboard 显示 pb文件网络结构

其他深度学习工具查看网络结构

Netron

安装完成后直接双击神经网络文件或者拖拽到,就能在netron中显示

左边为tensorboard 右边为netron显示的部分网络模型.

对比

netron 更加直观,可以显示更多的格式的网络模型. 比如hdf5文件等 keras 保存为h5 文件的时候需要使用save() 而不能用save_weight()

tensorboard 由于是tensorflow 官方的显示工具,可以显示模型的更多属性, 像netron 模型的有些属性没法显示.

转载请注明:veyvin » 机器学习模型的可视化

发表我的评论
取消评论
表情

Hi,您需要填写昵称和邮箱!

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址