Question

我用Keras实现了LSTM并在语料库上训练它。现在我想获取NN的输出并将其作为输入传递给另一个NN模型并在测试集上进行测试。我保存了NN的模型，但我不知道如何提取输出并将其提供给其他NN。我该怎么办？

这是我的代码：

model = Sequential()
model.add(Embedding(vocab_size, 50, input_length=seq_length))
model.add(LSTM(100, return_sequences=True))
model.add(LSTM(100))
model.add(Dense(100, activation='relu'))
model.add(Dense(vocab_size, activation='softmax'))
print(model.summary())
# compile model
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=
['accuracy'])
# fit model
model.fit(X, y, batch_size=128, epochs=500)
# save the model to file
model.save('model.h5')

Answer 1

无需保存模型即可提取输出。您只需使用predict()函数即可获得输出。在您的情况下，从模型中获取输出传递测试输入，如图所示

prediction = model.predict(X_test)

其中X_test是您的测试输入，'prediction'将包含您可以打印的输出。

要将此输出作为另一个模型的输入，首先创建另一个模型，然后将预测变量传递给model.fit()函数。

Answer 2

要在另一个模型中使用一个模型的输出，只需加载并使用它即可：

m0 = tf.keras.models.load_model("path_to_model_here", compile=False)
m0 = m0(some_input)

m1 = concatenate([some_input, m0])
m1 = Dense(DENSE_LAYER_NEURONS, input_shape=(INPUT_SIZE + M0_SIZE,), activation="relu")(m1)
m1 = Dense(DENSE_LAYER_NEURONS, activation="relu")(m1)
...

如何使用神经网络的输出作为Keras中另一个NN的输入

2 个答案: