Lstm 模型预测仅在 0 和 1 之间浮动

数据挖掘 Python 喀拉斯 张量流 lstm
2022-02-20 17:33:23

我创建了一个 LSTM 模型来预测几个月内的犯罪数量。对于每个月,我都会预测下个月的犯罪数量。目标是大于1的值(通常)但模型预测总是在0和1之间。我不明白这个问题。

model = Sequential()
model.add(LSTM(35, input_shape=(1, look_back)))
model.add(Dense(1))
model.summary()
model.compile(loss='mean_squared_error', optimizer='adam')
model.fit(trainX, trainY, epochs=20, batch_size=1, verbose=2)
testPredict = model.predict(testX)

这是使用此模型进行预测的示例,[0.5402554, 0.56099564, 0.5612484, 0.56098187, 0.5609817]但通常我应该有这样的东西[66.0, 20.0, 20.0, 10.0]

谢谢你的帮助

0个回答
没有发现任何回复~