cito (version 1.1)

plot.citodnn: Creates graph plot which gives an overview of the network architecture.

Description

Creates graph plot which gives an overview of the network architecture.

Usage

# S3 method for citodnn
plot(x, node_size = 1, scale_edges = FALSE, ...)
# S3 method for citodnnBootstrap
plot(x, node_size = 1, scale_edges = FALSE, which_model = 1, ...)

Value

A plot made with 'ggraph' + 'igraph' that represents the neural network

Arguments

x

a model created by dnn

node_size

size of node in plot

scale_edges

edge weight gets scaled according to other weights (layer specific)

...

no further functionality implemented yet

which_model

which model from the ensemble should be plotted

Examples

# \donttest{
if(torch::torch_is_installed()){
library(cito)

set.seed(222)
validation_set<- sample(c(1:nrow(datasets::iris)),25)

# Build and train  Network
nn.fit<- dnn(Sepal.Length~., data = datasets::iris[-validation_set,])

plot(nn.fit)
}
# }