def create_dataset(self, dataset): dataX, dataY = [] for i in range(len(dataset)-self.look_back): a = dataset[i:(i + self.look_back), 0] dataX.append(a) dataY.append(dataset[i + self.look_back, 0]) return np.array(dataX), np.array(dataY)