Keras中间层输出

2024-04-26 04:15:11 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在使用Keras的函数API时得到中间层的输出。我可以在使用标准顺序API时获得输出,但不能使用函数API。在

我在做这个玩具的例子:

from keras.models import Sequential
from keras.layers import Input, Dense,TimeDistributed
from keras.models import Model
from keras.layers import Dense, LSTM, Bidirectional,Masking

inputs = [[[0,0,0],[0,0,0],[0,0,0],[0,0,0]],[[1,2,3],[4,5,6],[7,8,9],[10,11,12]],[[10,20,30],[40,50,60],[70,80,90],[100,110,120]]]

model = Sequential()
model.add(Masking(mask_value=0., input_shape = (4,3)))
model.add(Bidirectional(LSTM(3,return_sequences = True),merge_mode='concat'))
model.add(TimeDistributed(Dense(3,activation = 'softmax')))


print "First layer:"
intermediate_layer_model = Model(input=model.input,output=model.layers[0].output)
print intermediate_layer_model.predict(inputs)
print ""
print "Second layer:"
intermediate_layer_model = Model(input=model.input,output=model.layers[1].output)
print intermediate_layer_model.predict(inputs)
print ""
print "Third layer:"
intermediate_layer_model = Model(input=model.input,output=model.layers[2].output)
print intermediate_layer_model.predict(inputs)

但是如果我使用函数式API,它就不起作用了。输出不正确。例如,它在第二层输出初始输入:

^{pr2}$

回答:好吧,我自己解决。显然,当使用函数API时,0层是输入本身。所以一切都向前移动了一个位置。在


Tags: 函数fromimportlayerapiinputoutputmodel