Add call to registration from the NeatGenomeFactory, preventing the e…

…nd user from needing to make a separate call.
colgreen · Jun 11, 2019 · 65f6d22 · 65f6d22
1 parent 85e671e
commit 65f6d22
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/SharpNeatLib/Genomes/Neat/NeatGenomeFactory.cs b/src/SharpNeatLib/Genomes/Neat/NeatGenomeFactory.cs
@@ -93,6 +93,12 @@ public NeatGenomeFactory(int inputNeuronCount, int outputNeuronCount)
 
             _genomeIdGenerator = new UInt32IdGenerator();
             _innovationIdGenerator = new UInt32IdGenerator();
+
+            if (neatGenomeParams.ActivationFn != null)
+            {
+                // Register the activation function with the library if it is specified.
+                ActivationFunctionRegistry.RegisterActivationFunction(neatGenomeParams.ActivationFn);
+            }
         }
 
         /// <summary>

diff --git a/src/SharpNeatLib/Network/ActivationFunctions/ActivationFunctionRegistry.cs b/src/SharpNeatLib/Network/ActivationFunctions/ActivationFunctionRegistry.cs
@@ -56,7 +56,10 @@ public static class ActivationFunctionRegistry
         /// </summary>
         public static void RegisterActivationFunction(IActivationFunction function)
         {
-            _registeredActivationFunctions.Add(function.FunctionId, function);
+            if (!_registeredActivationFunctions.ContainsKey(function.FunctionId))
+            {
+                _registeredActivationFunctions.Add(function.FunctionId, function);
+            }
         }
         #endregion // Static Public Methods
     }