regression: Regression problem

In mgoulet847/tagi: Tractable Approximate Gaussian Inference in Neural Networks

Description

This function trains neural network models to solve a regression problem.

Usage

regression(NN, x, y, trainIdx, testIdx)

Arguments

NN	Lists the structure of the neural network
x	Input data
y	Response data
trainIdx	Observations IDs that are assigned to the training set
testIdx	Observations IDs that are assigned to the testing set

Value

• Mean vector of parameters for each layer μ_{θ}

• Covariance matrix of parameters for each layer Σ_{θ}

• RMSE and LL metrics for each network models created

• Training time of each neural network models created

• Mean of predicted responses

• Variance of the predicted responses

mgoulet847/tagi documentation built on Dec. 21, 2021, 5:10 p.m.