diff --git a/src/HushGP/GP.hs b/src/HushGP/GP.hs index f8c7c21..f805a80 100644 --- a/src/HushGP/GP.hs +++ b/src/HushGP/GP.hs @@ -58,6 +58,7 @@ gpLoop' pushArgs generation evaluations population indexedTrainingData = do print "Incomplete Run, saving the best so far." | otherwise = gpLoop' pushArgs (succ generation) (evaluations + (populationSize pushArgs * length (fst $ trainingData pushArgs)) + (if generation `mod` downsampleParentsGens pushArgs == 0 then length parentReps * (length (fst indexedTrainingData) - length (fst $ trainingData pushArgs)) else 0) + (if bestIndPassesDownsample then length (fst indexedTrainingData) - length (fst $ trainingData pushArgs) else 0)) + (if elitism then ) nextAction where -- \| This will have downsampling added to it later.