Return fittest gene of entire run

Author: Rodrigo López Dato

shared/src/main/scala/rolodato/genetics/Gene.scala

@@ -42,3 +42,11 @@ trait Gene {
     }
   }
 }
+
+object Gene {
+  val DummyGene = new Gene {
+    def copy(newString: List[Int]): Gene = this
+    def fitness: Double = Double.NegativeInfinity
+    val string: List[Int] = List()
+  }
+}

shared/src/main/scala/rolodato/genetics/Genetic.scala

@@ -34,6 +34,7 @@ trait Genetic {
       var population = initial.to[ListBuffer]
       var fitnesses = ListBuffer[Double]()
       val selectionSize = (population.length * selectionPercentage).toInt
+      var best: Gene = Gene.DummyGene
       for (i <- 1 to iterations) {
         // Selection
         population = selection.selectPopulation(population, selectionSize).to[ListBuffer]
@@ -50,10 +51,16 @@ trait Genetic {
           population = population.updated(pos, mutation.mutate(population(pos)))
         }
         fitnesses += avgFitness(population)
+        best = {
+          val currentBest = population.maxBy(_.fitness)
+          if (currentBest.fitness > best.fitness) currentBest
+          else best
+        }
       }
       new GeneticResult {
         val finalPopulation = population.sortBy(-_.fitness).toList
         val fitnessEvolution = fitnesses.toList
+        val fittest = best
       }
     }
   }

shared/src/main/scala/rolodato/genetics/GeneticResult.scala

@@ -2,6 +2,6 @@ package rolodato.genetics
 
 trait GeneticResult {
   val finalPopulation: List[Gene]
-  def fittest: Gene = finalPopulation.sortBy(-_.fitness).head
+  val fittest: Gene
   val fitnessEvolution: List[Double]
 }