主页 / coding / 问题 / 77779141
Accepted
hasanghaforian
Asked: 2024-01-08 18:59:28 +0800 CST

LSTM 层输入的数据基数不明确

  • 772

假设我有这 7 个时间序列样本：

1,2,3,4,5,6,7

我知道每个样本与其之前的两个样本之间存在某种关系。这意味着当您知道两个较早的样本时1,2，您就可以预测下一个样本一定是3，并且2,3下一个样本是4，依此类推。

现在我想为上述样本训练RNN一个LSTM层。我所做的是：

import numpy as np
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers

X = np.array([[[1]],[[2]],[[3]],[[4]],[[5]],[[6]],[[7]]])
Y = np.array([[3],[4],[5],[6],[7]])

model = keras.Sequential([
    layers.LSTM(16, input_shape=(2, 1)),
    layers.Dense(1, activation="softmax")
])

model.compile(optimizer="rmsprop",
loss="mse",
metrics=["accuracy"])

model.fit(X, Y, epochs=1, batch_size=1)

但我遇到了这个错误：

ValueError: Data cardinality is ambiguous:
  x sizes: 7
  y sizes: 5
Make sure all arrays contain the same number of samples.

我不知道我要如何改变形状X并Y解决问题？

python

1 个回答

  1. Best Answer

    有几个问题。首先，由于监督培训，您需要确保巴黎培训（X和相应培训Y）的长度相同。接下来，由于您想要构建一个采用1和2并预测 的模型3，因此您还需要相应地准备数据加载器，以便它生成两个值 asX和一个目标值Y；例如，首先，它可能如下：X[0]: [[1], [2]]和Y[0]: [3]。最后，在最后一层中，您使用了激活，softmax但此处使用不正确，而应该是线性激活。下面是完整的工作代码。

    数据生成器

    data = np.array([1, 2, 3, 4, 5, 6, 7])
sequences_length = 2

def dataloader(data, sequences_length):
    X, Y = [], []
    for i in range(len(data) - sequences_length):
        X.append(data[i:i+sequences_length])
        Y.append(data[i+sequences_length])
    return np.array(X), np.array(Y)

X, Y = dataloader(data, sequences_length)
X = np.reshape(X, (X.shape[0], sequences_length , 1))

# check
for i in range(X.shape[0]):
    print(X[i].reshape(-1), Y[i])
[1 2] 3
[2 3] 4
[3 4] 5
[4 5] 6
[5 6] 7

    模型

    model = keras.Sequential([
    layers.LSTM(64, input_shape=(sequences_length, 1)),
    layers.Dense(1)
])
model.compile(optimizer="adam", loss="mse")
model.fit(X, Y, epochs=1000, batch_size=1)

    预言

    inference_data = np.array([[8, 9]]).reshape(
    1, sequences_length, 1
)
model.predict(inference_data)
1/1 [==============================] - 0s 25ms/step
array([[9.420095]], dtype=float32)
    • 1

相关问题