使用LSTM进行多变量预测单变量使用滑动窗口划分数据集(澳大利亚电力负荷数据)时,出现错误
https://blog.csdn.net/weixin_39653948/article/details/105446709
def create_dataset(X, y, seq_len=48):
features = []
targets = []
for i in range(0, len(X) - seq_len, 1):
data = X.iloc[i:i+seq_len] # 序列数据
label = y.iloc[i+seq_len] # 标签数据
# 保存到features和labels
features.append(data)
targets.append(label)
# 返回
return np.array(features),np.array(targets)
ValueError Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_196392/2181602682.py in <module>
1 # ① 构造训练特征数据集
2
----> 3 train_dataset, train_labels = create_dataset(X_train, y_train, seq_len=48)
~\AppData\Local\Temp/ipykernel_196392/485638540.py in create_dataset(X, y, seq_len)
13
14 # 返回
---> 15 return np.array(features),np.array(targets)
ValueError: cannot copy sequence with size 48 to array axis with dimension 5
想要完成数据集的划分