机器学习模型可视化

福斯基

于 2022-06-16 22:45:04 发布

阅读量476

点赞数

分类专栏：天文学数据 python 文章标签：机器学习 keras 深度学习

本文链接：https://blog.youkuaiyun.com/cool_bot/article/details/125324593

版权

该博客介绍了如何利用conda创建和管理虚拟环境，以及如何在环境中安装pydotplus和graphviz包。接着，展示了如何加载Keras模型，并使用keras.utils.vis_utils的plot_model函数生成模型的数据流图，以便于理解模型结构。示例中展示了Inception_v4模型的结构图。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先要安装conda虚拟环境环境，可以参考这里。

激活虚拟环境并安装以下两个包

conda install pydotplus
conda install graphviz

打开模型文件

from keras.utils.vis_utils import plot_model
model = InceV4.Incetion_v4(input_shape,outpt_shape)#你的机器学习模型实例
plot_model(model,to_file

最低0.47元/天解锁文章