def creat_dataset(dataset, look_back=1): dataX, dataY = [], [] for i in range(len(dataset)-look_back-1): a = dataset[i: (i+look_back)] dataX.append(a) dataY.append(dataset[i+look_back]) return np.array(dataX), np.array(dataY)解说
时间: 2023-12-25 19:02:35 浏览: 88
CreateDataSet:读取数据并创建数据集
这是一个用于创建时间序列数据集的函数,其中:
- `dataset`是原始时间序列数据
- `look_back`是每个输入序列的长度
函数将原始数据集转换为可以用于监督式学习的数据集。它通过滑动窗口的方式对原始数据进行分割,每个窗口的长度为`look_back`,并将窗口内的数据作为输入特征`dataX`,将窗口后面的一个数据作为输出标签`dataY`。函数返回的是包含输入特征`dataX`和输出标签`dataY`的numpy数组。
例如,如果原始数据集为[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],`look_back`为3,则函数将创建如下数据集:
```
dataX = [[1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6], [5, 6, 7], [6, 7, 8], [7, 8, 9]]
dataY = [4, 5, 6, 7, 8, 9, 10]
```
阅读全文