Dense: Create a Dense layer with 'in_size' inputs and 'out_size'...

In enweg/BFluxR: Implementation of Bayesian Neural Networks

Create a Dense layer with 'in_size' inputs and 'out_size' outputs using 'act' activation function

Description

Create a Dense layer with 'in_size' inputs and 'out_size' outputs using 'act' activation function

Usage

Dense(in_size, out_size, act = c("identity", "sigmoid", "tanh", "relu"))

Arguments

in_size	Input size
out_size	Output size
act	Activation function

Value

A list with the following content

• in_size - Input Size

• out_size - Output Size

• activation - Activation Function

• julia - Julia code representing the Layer

Examples

## Not run: 
  ## Needs previous call to `BayesFluxR_setup` which is time
  ## consuming and requires Julia and BayesFlux.jl
  BayesFluxR_setup(installJulia=TRUE, seed=123)
  net <- Chain(Dense(5, 5, "relu"))

## End(Not run)

enweg/BFluxR documentation built on Jan. 27, 2024, 6:43 p.m.