如何获得我的班级预测LSTM Keras

Question

我最近尝试构建一个程序，对Quora（问题对）数据集进行分类，无论它是否重复。我得到了基于真实y的准确性和损失，但IDK如何进行输出（预测y）任何人都可以帮助我吗？ 输出shud为1或0（二进制类）

这是句子合并代码，训练过程使用LSTM

merged = RNN(EMBED_HIDDEN_SIZE)(merged)

merged = layers.Dropout(dropoutp)(merged)
preds = layers.Dense(answer_size, activation='sigmoid')(merged)

model = Model([questiona, questionb], preds)

rmsprop = keras.optimizers.rmsprop(lr=lrn)
model.summary()

Answer 1

您可以通过将测试数据传递给预测函数来获取预测

predictions=model.predict(X)

Link to the docs