Plot Keras model

Source: R/plot_keras_model.R

Converts a Keras model to dot format and save to a file.

plot_keras_model(
  model,
  to_file = "model.png",
  show_shapes = FALSE,
  show_layer_names = TRUE,
  rankdir = "TB",
  expand_nested = FALSE,
  dpi = 96
)

Arguments

model

A Keras model instance
to_file

File name of the plot image.
show_shapes

whether to display shape information.
show_layer_names

whether to display layer names.
rankdir

`rankdir` argument passed to PyDot, a string specifying the format of the plot: 'TB' creates a vertical plot; 'LR' creates a horizontal plot.
expand_nested

Whether to expand nested models into clusters.
dpi

Dots per inch.

Value

saves a png image on the system and builds a plot in R