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