diff --git a/architectures/neon_components/general_network.py b/architectures/neon_components/general_network.py index 8bc9f7d..99ac6e9 100644 --- a/architectures/neon_components/general_network.py +++ b/architectures/neon_components/general_network.py @@ -42,6 +42,7 @@ class GeneralNeonNetwork(NeonArchitecture): 'tanh': neon.Tanh(), 'sigmoid': neon.Logistic(), 'elu': neon.Explin(), + 'selu': None, 'none': None } assert activation_function_string in activation_functions.keys(), \ diff --git a/architectures/tensorflow_components/general_network.py b/architectures/tensorflow_components/general_network.py index 5258976..60a0433 100644 --- a/architectures/tensorflow_components/general_network.py +++ b/architectures/tensorflow_components/general_network.py @@ -45,6 +45,7 @@ class GeneralTensorFlowNetwork(TensorFlowArchitecture): 'tanh': tf.nn.tanh, 'sigmoid': tf.nn.sigmoid, 'elu': tf.nn.elu, + 'selu': tf.nn.selu, 'none': None } assert activation_function_string in activation_functions.keys(), \