Merge pull request #1 from martiasG/feature/progressbar

martiasG · web-flow · commit 004dbdd3c21c · 2022-04-23T12:45:28.000-03:00
Update pygad.py
diff --git a/pygad.py b/pygad.py
@@ -4,6 +4,7 @@
 import pickle
 import time
 import warnings
+from tqdm.auto import tqdm
 
 class GA:
 
@@ -91,7 +92,10 @@ def __init__(self,
         on_parents: Accepts a function to be called after selecting the parents that mates. This function must accept 2 parameters: the first one represents the instance of the genetic algorithm and the second one represents the selected parents. Added in PyGAD 2.6.0.
         on_crossover: Accepts a function to be called each time the crossover operation is applied. This function must accept 2 parameters: the first one represents the instance of the genetic algorithm and the second one represents the offspring generated using crossover. Added in PyGAD 2.6.0.
         on_mutation: Accepts a function to be called each time the mutation operation is applied. This function must accept 2 parameters: the first one represents the instance of the genetic algorithm and the second one represents the offspring after applying the mutation. Added in PyGAD 2.6.0.
-        callback_generation: Accepts a function to be called after each generation. This function must accept a single parameter representing the instance of the genetic algorithm. If the function returned "stop", then the run() method stops without completing the other generations. Starting from PyGAD 2.6.0, the callback_generation parameter is deprecated and should be replaced by the on_generation parameter.
+        callback_generation: Accepts a function to be called after each generation. This function must accept a single parameter representing the instance of the genetic algorithm. If the function returned "stop", then the 
+        
+        
+        () method stops without completing the other generations. Starting from PyGAD 2.6.0, the callback_generation parameter is deprecated and should be replaced by the on_generation parameter.
         on_generation: Accepts a function to be called after each generation. This function must accept a single parameter representing the instance of the genetic algorithm. If the function returned "stop", then the run() method stops without completing the other generations. Added in PyGAD 2.6.0.
         on_stop: Accepts a function to be called only once exactly before the genetic algorithm stops or when it completes all the generations. This function must accept 2 parameters: the first one represents the instance of the genetic algorithm and the second one is a list of fitness values of the last population's solutions. Added in PyGAD 2.6.0. 
 
@@ -1201,7 +1205,7 @@ def run(self):
         if self.save_solutions:
             self.solutions.extend(self.population.copy())
 
-        for generation in range(self.num_generations):
+        for generation in tqdm(range(self.num_generations)):
             if not (self.on_fitness is None):
                 self.on_fitness(self, self.last_generation_fitness)
 
