bug fix

jaswinder9051998 · Sep 12, 2021 · 0f1e972 · 0f1e972
1 parent a4ac3df
commit 0f1e972
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 11 deletions.
diff --git a/zoofs/baseoptimizationalgorithm.py b/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):

diff --git a/zoofs/dragonflyoptimization.py b/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

diff --git a/zoofs/geneticoptimization.py b/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

diff --git a/zoofs/gravitationaloptimization.py b/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

diff --git a/zoofs/greywolfoptimization.py b/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

diff --git a/zoofs/particleswarmoptimization.py b/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