Dinfio Documentation — Function Reference: neural_network::set_activation_function()

Module neural → Class neural_network → set_activation_function()

Description

Set activation function \(A\) to the layer \(L_i\)

neural_network::set_activation_function(layer, f)

Parameters

layer (number) — The layer \(L_i\)
f (number) — The activation function \(A\). Please see activation_function for the activation functions list

Return Value

boolean — Always returns true

Usage Example

import neural
 
model = neural_network([784, 20, 10, 10], 0.01)
model.set_activation_function(1, activation.relu)