history <- model %>% fit(X_train, y_train, epochs = 50, batch_size = 128, validation_split = 0.15)