【发布时间】:2021-01-17 18:37:39
【问题描述】:
我已经构建了一个 LSTM 模型(见下文)并对其进行了训练。我的损失函数是二进制交叉熵,因为我正在做二进制分类。训练 y 数据是一组 0 和 1。
当我运行model.predict(x_test_scaled) 时,我得到一组值介于 0 和 1 之间的单个系列。我猜这是一个概率,但它是输出 = 0 还是输出 = 1 的概率?
model = Sequential()
model.add(LSTM(units=512, input_shape = (X_train.shape[1],X_train.shape[2]), return_sequences=True))
model.add(Dropout(0.3))
model.add(LSTM(units=512, return_sequences=False))
model.add(Dropout(0.3))
model.add(Dense(264), activation = 'tanh')
model.add(Dense(1))
【问题讨论】:
-
请从intro tour 重复on topic 和how to ask。 Stack Overflow 并不打算取代现有的教程和文档。关于 Keras 和建模的各种文档通常描述了如何解释输出。我们希望您已经阅读了足够多的内容,以便针对该知识提出具体问题。
-
我已阅读此网站上的文档。 link 它实际上并没有涵盖如何解释输出。它确实说“为输入样本生成输出预测。”,但这并没有真正的帮助。
-
这就是为什么我向您推荐了不止一份文件。您需要了解您训练的特定类别的模型会产生什么。
-
我很高兴阅读文档,如果您知道此信息,我将不胜感激包含此信息的文档链接。
-
你有发帖指南的链接吗?
标签: python tensorflow keras sequential