Created by: pkuyym

'X': np.random.rand(128, 256).astype(np.float64),
'Y': np.random.rand(128, 1).astype(np.float64)