Allow for different link functions to be swapped in for the hidden layer