[1113] | 1 | from evolalg.base.frams_step import FramsStep |
---|
| 2 | from evolalg.mutation_cross.frams_cross import FramsCross |
---|
| 3 | from evolalg.mutation_cross.frams_mutation import FramsMutation |
---|
| 4 | import random |
---|
| 5 | |
---|
| 6 | class FramsCrossOrMutate(FramsStep): |
---|
| 7 | def __init__(self, frams_lib, commands, cross_prob, mutate_prob, mutate_commands=None, cross_commands=None): |
---|
| 8 | super().__init__(frams_lib, commands) |
---|
| 9 | |
---|
| 10 | self.cross_prob = cross_prob |
---|
| 11 | self.mutate_prob = mutate_prob |
---|
| 12 | |
---|
| 13 | self.frams_mutate = FramsMutation(frams_lib, mutate_commands, 1) |
---|
| 14 | self.frams_cross = FramsCross(frams_lib, cross_commands, 1) |
---|
| 15 | |
---|
| 16 | def call(self, population): |
---|
| 17 | mutate_idx = [] |
---|
| 18 | cross_idx = [] |
---|
| 19 | for i in range(len(population)): |
---|
| 20 | rng = random.random() |
---|
| 21 | if rng < self.cross_prob: |
---|
| 22 | cross_idx.append(i) |
---|
| 23 | elif rng < self.cross_prob+self.mutate_prob: |
---|
| 24 | mutate_idx.append(i) |
---|
| 25 | |
---|
| 26 | mutate_ind = [population[_] for _ in mutate_idx] |
---|
| 27 | cross_ind = [population[_] for _ in cross_idx] |
---|
| 28 | |
---|
| 29 | if len(mutate_ind) > 0: |
---|
| 30 | self.frams_mutate(mutate_ind) |
---|
| 31 | if len(cross_idx) > 0: |
---|
| 32 | self.frams_cross(cross_idx) |
---|
| 33 | |
---|
| 34 | for i,ind in zip(mutate_idx, mutate_ind): |
---|
| 35 | population[i] = ind |
---|
| 36 | |
---|
| 37 | for i,ind in zip(cross_idx, cross_ind): |
---|
| 38 | population[i] = ind |
---|
| 39 | |
---|
| 40 | return population |
---|