Added FitnessBased PoolGenerator and started using it in Program.cs

Author: BotMaster3000

NeuralBotMasterFramework/NeuralBotMasterFramework/Logic/PoolGenerators/FitnessBasedPoolGenerator.cs

@@ -0,0 +1,43 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using NeuralBotMasterFramework.Helper;
+using NeuralBotMasterFramework.Interfaces;
+
+namespace NeuralBotMasterFramework.Logic.PoolGenerators
+{
+    public class FitnessBasedPoolGenerator : IBreedingPoolGenerator
+    {
+        public List<IWeightedNetwork> GenerateBreedingPool(Dictionary<IWeightedNetwork, double> networksAndFitness)
+        {
+            List<IWeightedNetwork> poolList = new List<IWeightedNetwork>();
+            double[] fitnesses = MultiplyUntilBiggerThanOne(networksAndFitness.Values.ToArray());
+            int totalFitness = (int)Math.Round(fitnesses.Sum(), 0, MidpointRounding.AwayFromZero);
+            for (int networkIndex = 0; networkIndex < networksAndFitness.Count; ++networkIndex)
+            {
+                int totalNetworksToAdd = (int)Math.Round(fitnesses[networkIndex] / totalFitness * networksAndFitness.Count * 100);
+                KeyValuePair<IWeightedNetwork, double> currentNetworkAndFitness = networksAndFitness.ElementAt(networkIndex);
+                for (int fitness = 0; fitness < totalNetworksToAdd; ++fitness)
+                {
+                    poolList.Add(currentNetworkAndFitness.Key);
+                }
+            }
+            return poolList;
+        }
+
+        private double[] MultiplyUntilBiggerThanOne(double[] values)
+        {
+            double[] returnValues = (double[])values.Clone();
+            while (returnValues.Min() < 1)
+            {
+                for (int i = 0; i < returnValues.Length; ++i)
+                {
+                    returnValues[i] *= 10;
+                }
+            }
+            return returnValues;
+        }
+    }
+}

NeuralBotMasterFramework/NeuralBotMasterFramework/NeuralBotMasterFramework.csproj

@@ -53,6 +53,7 @@
     <Compile Include="Interfaces\INode.cs" />
     <Compile Include="Logic\Algorithms\GeneticAlgorithm.cs" />
     <Compile Include="Logic\Networks\WeightedNetwork.cs" />
+    <Compile Include="Logic\PoolGenerators\FitnessBasedPoolGenerator.cs" />
     <Compile Include="Logic\PoolGenerators\IndexBasedPoolGenerator.cs" />
     <Compile Include="Models\Layer.cs" />
     <Compile Include="Models\Node.cs" />

NeuralBotMasterFramework/NeuralBotMasterFrameworkTests/Logic/PoolGenerators/FitnessBasedPoolGeneratorTests.cs

@@ -0,0 +1,47 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using NeuralBotMasterFramework.Logic.PoolGenerators;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using NeuralBotMasterFramework.Logic.Algorithms;
+using NeuralBotMasterFramework.Helper;
+using NeuralBotMasterFramework.Interfaces;
+
+namespace NeuralBotMasterFramework.Logic.PoolGenerators.Tests
+{
+    [TestClass]
+    public class FitnessBasedPoolGeneratorTests
+    {
+        [TestMethod]
+        public void GenerateBreedingPoolTest()
+        {
+            const int TOTAL_NETWORKS = 15;
+            const int INPUT_NODES = 5;
+            const int HIDDEN_NODES = 5;
+            const int HIDDEN_LAYERS = 5;
+            const int OUTPUT_NODES = 1;
+            const int NETWORKS_TO_KEEP = 10;
+            GeneticAlgorithm algorithm = new GeneticAlgorithm(TOTAL_NETWORKS, INPUT_NODES, HIDDEN_NODES, HIDDEN_LAYERS, OUTPUT_NODES);
+            double[][] inputData = new double[][] { GetRandomDoubleArray(INPUT_NODES) };
+            double[][] expectedData = new double[][] { GetRandomDoubleArray(OUTPUT_NODES) };
+            algorithm.SetupTest(inputData, expectedData);
+            algorithm.PropagateAllNetworks();
+            algorithm.SortByFitness();
+            FitnessBasedPoolGenerator generator = new FitnessBasedPoolGenerator();
+            List<IWeightedNetwork> networkList = generator.GenerateBreedingPool(algorithm.NetworksAndFitness.Take(NETWORKS_TO_KEEP).ToDictionary(x => x.Key, x => x.Value));
+            Assert.IsTrue(networkList.Count > 0);
+        }
+
+        private double[] GetRandomDoubleArray(int totalValues)
+        {
+            double[] array = new double[totalValues];
+            for (int i = 0; i < array.Length; ++i)
+            {
+                array[i] = RandomNumberGenerator.GetNextDouble();
+            }
+            return array;
+        }
+    }
+}

NeuralBotMasterFramework/NeuralBotMasterFrameworkTests/NeuralBotMasterFrameworkTests.csproj

@@ -58,6 +58,7 @@
   <ItemGroup>
     <Compile Include="Helper\RandomNumberGeneratorTests.cs" />
     <Compile Include="Logic\Algorithms\GeneticAlgorithmTests.cs" />
+    <Compile Include="Logic\PoolGenerators\FitnessBasedPoolGeneratorTests.cs" />
     <Compile Include="Logic\WeightedNetworkTests.cs" />
     <Compile Include="Models\LayerTests.cs" />
     <Compile Include="Models\NodeTests.cs" />

NeuralBotMasterFramework/TestConsole/Program.cs

@@ -4,6 +4,7 @@
 using System.Text;
 using System.Threading.Tasks;
 using NeuralBotMasterFramework.Logic.Algorithms;
+using NeuralBotMasterFramework.Logic.PoolGenerators;
 
 namespace TestConsole
 {
@@ -23,13 +24,16 @@ static void Main(string[] args)
             int hiddenLayers = 3;
             int outputNodes = totalNumberLength;
 
-            int networksToKeep = 10;
+            int networksToKeep = 100;
             double mutationRate = 0.2;
             double mutationChance = 0.4;
 
             SetupBinaryData();
 
-            GeneticAlgorithm algorithm = new GeneticAlgorithm(totalNetworks, inputNodes, hiddenNodes, hiddenLayers, outputNodes);
+            GeneticAlgorithm algorithm = new GeneticAlgorithm(totalNetworks, inputNodes, hiddenNodes, hiddenLayers, outputNodes)
+            {
+                PoolGenerator = new IndexBasedPoolGenerator()
+            };
 
             algorithm.SetupTest(inputData, expectedData);