SEMdeep (version 0.1.0)

nplot: Create a plot for a neural network model

Description

The function draws a neural network plot as a neural interpretation diagram using with the plotnet function of the NeuralNetTools R package.

Usage

nplot(dnn.fit, bias = FALSE, ...)

Value

nplot returns invisibly the graphical object representing the neural network architecture of NeuralNetTools.

Arguments

dnn.fit

A neural network model from cito R package.

bias

A logical value, indicating whether to draw biases in the layers (default = FALSE).

...

Currently ignored.

Author

Mario Grassi

Details

The induced subgraph of the input graph mapped on data variables. Based on the estimated connection weights, if the connection weight W > 0, the connection is activated and it is highlighted in red; if W < 0, the connection is inhibited and it is highlighted in blue.

References

Beck, M.W. 2018. NeuralNetTools: Visualization and Analysis Tools for Neural Networks. Journal of Statistical Software. 85(11):1-20.

Examples

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

# load ALS data
ig<- alsData$graph
data<- alsData$exprs
data<- transformData(data)$data

dnn0 <- SEMdnn(ig, data, train=1:nrow(data), grad = FALSE,
			#loss = "mse", hidden = 5*K, link = "selu",
			loss = "mse", hidden = c(10, 10, 10), link = "selu",
			validation = 0, bias = TRUE, lr = 0.01,
			epochs = 32, device = "cpu", verbose = TRUE)

 for (j in 1:length(dnn0$model)) { 
   nplot(dnn0$model[[j]])
   Sys.sleep(5)
 }
}
# }