athena__initialiser Module

Module containing functions to set up initialisers

This module contains functions to set up initialisers for the weights and biases of a neural network model Examples of initialsers in keras: https://keras.io/api/layers/initializers/


Uses


Functions

public function get_default_initialiser(activation, is_bias) result(name)

Get the default initialiser based on the activation function

Arguments

Type IntentOptional Attributes Name
character(len=*), intent(in) :: activation

Activation function
logical, intent(in), optional :: is_bias

Boolean whether initialiser is for bias

Return Value character(len=:), allocatable

public function initialiser_setup(input, error) result(initialiser)

Set up the initialiser function

Arguments

Type IntentOptional Attributes Name
class(*) :: input

Name of initialiser or initialiser object
integer, intent(out), optional :: error

Error code

Return Value class(base_init_type), allocatable

Initialiser function