输入不用修改,输出Y改为双输出
时间: 2024-03-20 14:40:21 浏览: 14
好的,修改后的函数如下所示:
```python
def _slice_multi(self):
# self.n 是数据集总行数
# self.m 是列
s = self.ps * self.skip + self.Ck - 1 + self.h - 1
X1 = np.zeros((self.n - s, self.w, self.m)) # 短期数据集
X2 = np.zeros((self.n - s, self.ps * self.Ck, self.m)) # 长期数据集
Y1 = np.zeros((self.n - s, self.m)) # 输出数据集1
Y2 = np.zeros((self.n - s, self.m)) # 输出数据集2
for i in range(s, self.n):
t = i - self.h + 1
X1[i - s] = self.raw[t - self.w:t].copy()
idx = []
for k in range(self.ps):
# 执行self.ps次,每次间隔self.skip长,取self.Ck=6, total = self.ps * self.Ck
idx = list(range(t - self.Ck - k * self.skip, t - k * self.skip)) + idx
idx = np.array(idx, dtype=int)
X2[i - s] = self.raw[idx].copy()
Y1[i - s] = self.raw[i].copy()
Y2[i - s] = self.raw[i+1].copy()
# 返回特征和标签
return (X1, X2), (Y1, Y2)
```
修改后的函数输出了两个元组,每个元组包含了两个数组,分别对应两个输出。这样的修改可以满足双输出的需求。