获取keras中间层输出、模型保存与加载

本文主要是介绍获取keras中间层输出、模型保存与加载，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1. 获取keras中间层输出

# model summary and plot
import keras
from keras.models import Model
from keras.utils  import plot_modelDocEncoder.summary()
DocEncoder.plot_model()model.summary()
model.plot_model()

# get mid-output from parent-graph 
from keras import backend as Kattn_layer = K.function(inputs=model.input, outputs=model.layers[-4].output)
attn_layer_output = attn_layer([input_])[0]# get mid-output from children-graph
att_layer = K.function(inputs=DocEncoder.get_input_at(0), outputs=DocEncoder.layers[-2].output)
att_layer_output = att_layer([input_[0]])[0]

 

2. keras模型的保存与加载

import keras
from keras.models import Model
from keras.models import load_modelmodel = Model(input, output)# save
model.save('my_model.h5')# load
model = load_model('my_model.h5')

 

这篇关于获取keras中间层输出、模型保存与加载的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---