Use option.populationSize instead of size variable

ai · Jan 4, 2011 · 712b98b · 712b98b
1 parent e5d99ac
commit 712b98b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/drivers/genetic.js b/lib/drivers/genetic.js
@@ -14,7 +14,6 @@ init(function(name, initMsg) {
         var bestFitness    = undefined
         var nextPopulation = []
         var processing     = false
-        var size           = 0
         var generation     = 0
         var jobs           = 0
 
@@ -29,12 +28,12 @@ init(function(name, initMsg) {
             population = []
             if ("undefined" == typeof(options.generator)) {
                 population = options.population
+                options.populationSize = population.length
             } else {
                 for (var i = 0; i < options.populationSize; i++) {
                     population.push([options.generator(i), null])
                 }
             }
-            size = population.length
         }
 
         function processNext(to) {
@@ -44,7 +43,7 @@ init(function(name, initMsg) {
                 nextPopulation.sort(function(a, b) {
                     return options.compare(a[1], b[1])
                 })
-                population = nextPopulation.slice(0, size)
+                population = nextPopulation.slice(0, options.populationSize)
                 nextPopulation = []
                 generation++