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

【问题标题】：How to get my class prediction LSTM Keras如何获得我的班级预测 LSTM Keras
【发布时间】：2018-03-31 19:55:44
【问题描述】：

我最近尝试构建一个程序，对 Quora（问题对）数据集进行分类，无论它是否重复。我得到了基于真实 y 的准确性和损失，但是 IDK 如何进行输出（预测的 y）有人可以帮助我吗？输出应为 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()

【问题讨论】：

标签： python tensorflow keras lstm rnn

【解决方案1】：

您可以通过将测试数据传递给预测功能来获得预测

predictions=model.predict(X)

Link to the docs

【讨论】：

是的，谢谢 ^^ 我得到了输出，但是你知道为什么输出是浮点数，不是像 1 或 0 ._.
将最后一层的激活改为softmax