Ensure offspring new values are within variable limits

diogomatoschaves · Mar 8, 2022 · 86279e9 · 86279e9
1 parent a0015b4
commit 86279e9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/geneal/genetic_algorithms/_continuous.py b/geneal/genetic_algorithms/_continuous.py
@@ -143,10 +143,15 @@ def create_offspring(
                 first_parent[crossover_pt] - sec_parent[crossover_pt]
             )
         else:
+            variable_limits = self.variables_limits[crossover_pt]
+
             p_new = first_parent[crossover_pt] - np.round(
                 beta * (first_parent[crossover_pt] - sec_parent[crossover_pt])
             )
 
+            if not variable_limits[0] <= p_new <= variable_limits[1]:
+                p_new = np.random.randint(variable_limits[0], variable_limits[1] + 1)
+
         return np.hstack(
             (first_parent[:crossover_pt], p_new, sec_parent[crossover_pt + 1 :])
         )