bug fix

Author: jaswinder9051998

zoofs/baseoptimizationalgorithm.py

@@ -10,7 +10,7 @@ class BaseOptimizationAlgorithm(ABC):
 
     def __init__(self,
                  objective_function,
-                 n_iteration=50,
+                 n_iteration: int = 1000,
                  timeout: int = None,
                  population_size=50,
                  minimize=True):

zoofs/dragonflyoptimization.py

@@ -18,7 +18,7 @@ class DragonFlyOptimization(BaseOptimizationAlgorithm):
 
     def __init__(self,
                  objective_function,
-                 n_iteration=50,
+                 n_iteration: int = 1000,
                  timeout: int = None,
                  population_size=50,
                  minimize=True):
@@ -31,7 +31,7 @@ def __init__(self,
         population_size: int, default=50
             Total size of the population 
 
-        n_iteration: int, default=50
+        n_iteration: int, default=1000
             Number of time the Optimization algorithm will run
 
         timeout: int = None

zoofs/geneticoptimization.py

@@ -17,7 +17,7 @@ class GeneticOptimization(BaseOptimizationAlgorithm):
 
     def __init__(self,
                  objective_function,
-                 n_iteration=20,
+                 n_iteration: int = 1000,
                  timeout: int = None,
                  population_size=20,
                  selective_pressure=2,
@@ -30,7 +30,7 @@ def __init__(self,
         objective_function : user made function of the signature 'func(model,X_train,y_train,X_test,y_test)'
             The function must return a value, that needs to be minimized/maximized.
 
-        n_iteration : int, default=50
+        n_iteration : int, default=1000
             Number of time the Optimization algorithm will run
 
         timeout: int = None

zoofs/gravitationaloptimization.py

@@ -18,7 +18,7 @@ class GravitationalOptimization(BaseOptimizationAlgorithm):
 
     def __init__(self,
                  objective_function,
-                 n_iteration=50,
+                 n_iteration: int = 1000,
                  timeout: int = None,
                  population_size=50,
                  g0=100,
@@ -30,7 +30,7 @@ def __init__(self,
         objective_function : user made function of the signature 'func(model,X_train,y_train,X_test,y_test)'
             The function must return a value, that needs to be minimized/maximized.
 
-        n_iteration : int, default=50
+        n_iteration : int, default=1000
             Number of time the Optimization algorithm will run
 
         timeout: int = None

zoofs/greywolfoptimization.py

@@ -19,7 +19,7 @@ class GreyWolfOptimization(BaseOptimizationAlgorithm):
 
     def __init__(self,
                  objective_function,
-                 n_iteration=50,
+                 n_iteration: int = 1000,
                  timeout: int = None,
                  population_size=50,
                  minimize=True):
@@ -29,7 +29,7 @@ def __init__(self,
         objective_function : user made function of the signature 'func(model,X_train,y_train,X_test,y_test)'
             The function must return a value, that needs to be minimized/maximized.
 
-        n_iteration : int, default=50
+        n_iteration : int, default=1000
             Number of time the Optimization algorithm will run
 
         timeout: int = None

zoofs/particleswarmoptimization.py

@@ -18,7 +18,7 @@ class ParticleSwarmOptimization(BaseOptimizationAlgorithm):
 
     def __init__(self,
                  objective_function,
-                 n_iteration: int = 50,
+                 n_iteration: int = 1000,
                  timeout: int = None,
                  population_size=50,
                  minimize=True,
@@ -34,7 +34,7 @@ def __init__(self,
         population_size: int, default=50
             Total size of the population , default=50
 
-        n_iteration: int = 50
+        n_iteration: int, default=1000
             Number of time the Particle Swarm Optimization algorithm will run
 
         timeout: int = None