X_gen, y_gen = None, None