我已经训练了一个keras模型,并将其保存以供以后进行预测。但是,我使用以下方法加载了保存的模型:
from keras.models import load_model
#Restore saved keras model
restored_keras_model = load_model("C:/*******/saved_model.hdf5")
现在,我想保存已加载模型的图像,以便在进行预测之前可以在使用之前对其进行可视化。
在keras中有没有办法做到这一点,还是需要使用另一个库?
答案 0 :(得分:2)
是的,除了执行restore_keras_model.summary()之外,您还可以使用plot_model API将模型架构另存为png文件。
from keras.utils import plot_model
plot_model(restored_keras_model, to_file='model.png')