plot_keras_model {kerastuneR}

Plot Keras model

Description

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

Usage

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