X_train, y_train = generate(250, 0.15)
X_test, y_test = generate(500, 0.15)