LBBNN (version 0.1.2)

summary.LBBNN_Net: Summary of LBBNN fit

Description

Summary method for objects of the LBBNN_Net class. Only applies to objects trained with input_skip = TRUE.

Usage

# S3 method for LBBNN_Net
summary(object, ...)

Value

A data.frame containing the above information. The function prints a formatted summary to the console. The returned data.frame is invisible.

Arguments

object

An object of class LBBNN_Net.

...

further arguments passed to or from other methods.

Details

The returned table combines two types of information:

• Number of times each input variable is included in the active paths from each layer (obtained from get_input_inclusions()).

• Average inclusion probabilities for each input variable from each layer, including a final column showing the average across all layers.

Examples

# \donttest{ 
x<-torch::torch_randn(3,2) 
b <- torch::torch_rand(2)
y <- torch::torch_matmul(x,b)
train_data <- torch::tensor_dataset(x,y)
train_loader <- torch::dataloader(train_data,batch_size = 3,shuffle=FALSE)
problem<-'regression'
sizes <- c(2,1,1) 
inclusion_priors <-c(0.9,0.2) 
inclusion_inits <- matrix(rep(c(-10,10),2),nrow = 2,ncol = 2)
stds <- c(1.0,1.0)
model <- LBBNN_Net(problem,sizes,inclusion_priors,stds,inclusion_inits,flow = FALSE,
input_skip = TRUE)
train_LBBNN(epochs = 1,LBBNN = model, lr = 0.01,train_dl = train_loader)
summary(model)
# }