plot.citodnn {cito} | R Documentation |
Creates graph plot which gives an overview of the network architecture.
## S3 method for class 'citodnn'
plot(x, node_size = 1, scale_edges = FALSE, ...)
x |
a model created by |
node_size |
size of node in plot |
scale_edges |
edge weight gets scaled according to other weights (layer specific) |
... |
no further functionality implemented yet |
A plot made with 'ggraph' + 'igraph' that represents the neural network
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)
}