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中间层输出、模型保存与加载的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。