Hoftix tournament selection (#956)

aimclub · Oct 24, 2022 · 91c5ecf · 91c5ecf
1 parent 81e46c6
commit 91c5ecf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fedot/core/optimisers/gp_comp/operators/selection.py b/fedot/core/optimisers/gp_comp/operators/selection.py
@@ -71,7 +71,7 @@ def tournament_selection(individuals: PopulationT, pop_size: int, fraction: floa
     for _ in range(iterations_limit):
         if len(chosen) >= pop_size:
             break
-        group = sample(individuals, group_size)
+        group = sample(individuals, min(group_size, len(individuals)))
         best = max(group, key=lambda ind: ind.fitness)
         individuals.remove(best)
         chosen.append(best)