Plot CISS-VAE architecture schematic

Usage

plot_cissvae_arch(
  model_or_arch,
  title = NULL,
  color_shared = "skyblue",
  color_unshared = "lightcoral",
  color_input = "lightgreen",
  color_latent = "gold",
  color_output = "lightgreen",
  layer_gap = 2,
  phase_gap = 0.5,
  y_gap = 2,
  box_w = 1.2,
  box_h = 1
)

Arguments

model_or_arch: Python CISSVAE object (reticulate)
title: Optional title
color_shared: Fill for shared layers
color_unshared: Fill for unshared layers
color_input: Fill override for Input
color_latent: Fill override for Latent
color_output: Fill override for Output
layer_gap: Horizontal gap between layers in the same phase
phase_gap: Horizontal gap between phases
y_gap: Vertical gap between cluster rows
box_w: Box width
box_h: Box height for unshared layers