Blog2.R

use_session_with_seed(1,disable_parallel_cpu = FALSE)
model = keras_model_sequential()
model %>%
  layer_dense(input_shape = ncol(x), units = 10, activation = "relu") %>%
  layer_dropout(0.2) %>%
  layer_dense(units = 10, activation = "relu") %>%
  layer_dropout(0.2) %>%
  layer_dense(units = 3, activation = "softmax")
model %>%
  compile(
    loss = "categorical_crossentropy",
    optimizer = "adagrad",
    metrics = "accuracy"
)
fit = model %>%
  fit(
    x = x,
    y = y,
    shuffle = T,
    validation_split = 0.3,
    epochs = 200,
    batch_size = 5
)
plot(fit)