in src/run.py [0:0]
def load_data(data_path):
trX = np.load(f'{data_path}_trX.npy')
trY = np.load(f'{data_path}_trY.npy')
vaX = np.load(f'{data_path}_vaX.npy')
vaY = np.load(f'{data_path}_vaY.npy')
teX = np.load(f'{data_path}_teX.npy')
teY = np.load(f'{data_path}_teY.npy')
return (trX, trY), (vaX, vaY), (teX, teY)