Skip to contents

Plot the CNN architecture

Usage

# S3 method for citoarchitecture
plot(x, input_shape, output_shape, ...)

Arguments

x

an object of class citoarchitecture created by create_architecture

input_shape

a vector with the dimensions of a single sample (e.g. c(3,28,28))

output_shape

the number of nodes in the output layer

...

additional arguments

Value

nothing