def __init__(self, X, y=None, batch_size=32):
    self.X = X
    self.y = y
    self.batch_size = batch_size
def __init__(self, X, y=None, batch_size=32):
    self.X = X
    self.y = y
    self.batch_size = batch_size