1  import numpy as np 
   2  
   3  def new_dataset(dataset, step_size):
   4  	data_X, data_Y = [], []
   5  	for i in range(len(dataset)-step_size-1):
   6  		a = dataset[i:(i+step_size), 0]
   7  		data_X.append(a)
   8  		data_Y.append(dataset[i + step_size, 0])
   9  	return np.array(data_X), np.array(data_Y)
  10