bugfix

EpistasisLab · Jun 28, 2023 · 2e17842 · 2e17842
1 parent 4724480
commit 2e17842
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tpot2/evolvers/steady_state_evolver.py b/tpot2/evolvers/steady_state_evolver.py
@@ -417,8 +417,8 @@ def optimize(self):
                         if len(parents_df) < 2:
                             var_ops = ["mutate" for _ in range(n_individuals_to_submit)]
                         else:
-                            var_ops = [np.random.choice(["crossover","mutate_then_crossover","crossover_then_mutate",'mutate']) for _ in range(n_individuals_to_submit)]
-                        
+                            var_ops = [np.random.choice(["crossover","mutate_then_crossover","crossover_then_mutate",'mutate'],p=[self.crossover_probability,self.mutate_then_crossover_probability, self.crossover_then_mutate_probability,self.mutate_probability]) for _ in range(n_individuals_to_submit)]
+
                         parents = []
                         for op in var_ops:
                             if op == "mutate":