Plot visualization in R
model %>% fit (x = x_train, y = y_train, epochs = 25, batch_size = 32, validation_split =.2