From 1375975e9c9c56b6e484cef86b6e2be33bc32044 Mon Sep 17 00:00:00 2001 From: morrisnein Date: Mon, 24 Oct 2022 13:51:13 +0300 Subject: [PATCH] adapt size of selection in loop --- fedot/core/optimisers/gp_comp/operators/selection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fedot/core/optimisers/gp_comp/operators/selection.py b/fedot/core/optimisers/gp_comp/operators/selection.py index 79df4e9327..5d369ef6d5 100644 --- 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)