(dataset, lookback, predict_time)
| 26 | |
| 27 | # ---------data transform----------- |
| 28 | def create_rnn_data(dataset, lookback, predict_time): |
| 29 | x = [] |
| 30 | y = [] |
| 31 | for i in range(len(dataset) - lookback - predict_time): |
| 32 | x.append(dataset[i:i + lookback]) |
| 33 | y.append(dataset[i + lookback + predict_time - 1]) |
| 34 | return np.array(x), np.array(y) |
| 35 | |
| 36 | |
| 37 | def get_a_delta(adj): # D^-1/2 * A * D^-1/2 |