keras-model-plot

keras <- reticulate::import(module = "keras")
model <- keras$models$Sequential(
  list(
    keras$layers$Dense(output_dim = 32L, input_dim = 784L),
    keras$layers$Activation(activation = "relu"),
    keras$layers$Dense(output_dim = 10),
    keras$layers$Activation(activation = "softmax")
  )
)

DiagrammeR::grViz(
  diagram = keras$utils$visualize_util$model_to_dot(
    model = model, show_shapes = TRUE
  )$create(prog = "dot", format = "dot")
)