source: framspy/evolalg_steps/selection/tournament.py @ 1185

Last change on this file since 1185 was 1185, checked in by Maciej Komosinski, 17 months ago

Renamed a module; new name is "evolalg_steps"

File size: 671 bytes
Line 
1from evolalg_steps.base.individual import Individual
2from typing import List
3import random
4
5from evolalg_steps.base.step import Step
6from evolalg_steps.selection.selection import Selection
7
8
9class TournamentSelection(Selection):
10    def __init__(self, tournament_size: int, fit_attr="fitness", copy=False, *args, **kwargs):
11        super(TournamentSelection, self).__init__(copy, *args, **kwargs)
12        self.tournament_size = tournament_size
13        self.fit_attr = fit_attr
14
15    def select_next(self, population):
16        selected = [random.choice(population) for i in range(self.tournament_size)]
17        return max(selected, key=lambda x: getattr(x, self.fit_attr))
Note: See TracBrowser for help on using the repository browser.