activationFunIndex: Assign ID to activation functions

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

Description

This function assigns an ID number depending on the type of activation function.

Usage

activationFunIndex(funName)

Arguments

funName

Type of activation function: "tanh", "sigm", "cdf", "relu" or "softplus"

Value

An ID number which corresponds to:

• 1 if funName is "tanh"

• 2 if funName is "sigm"

• 3 if funName is "cdf"

• 4 if funName is "relu"

• 5 if funName is "softplus"

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