Total Complexity | 5 |
Total Lines | 25 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | """Contains WholeGroupPso class.""" |
||
7 | class OptimizationController(object): |
||
8 | @staticmethod |
||
9 | def should_continue(stages): |
||
10 | for stage in stages: |
||
11 | if OptimizationStatus.failed == stage.optimization_status: |
||
12 | return False |
||
13 | for stage in stages: |
||
14 | if OptimizationStatus.success != stage.optimization_status: |
||
15 | return True |
||
16 | return False |
||
17 | |||
18 | class WholeGroupPso(object): |
||
19 | """Optimize whole group.""" |
||
20 | |||
21 | def __init__(self, process): |
||
22 | self.process = process |
||
23 | self.number_of_particles = 40 |
||
24 | |||
25 | def optimize(self, ordered_stages_to_optimize): |
||
26 | swarm = Swarm(self.process, ordered_stages_to_optimize, |
||
27 | self.number_of_particles) |
||
28 | swarm.initialize() |
||
29 | while OptimizationController.should_continue( |
||
30 | ordered_stages_to_optimize): |
||
31 | swarm.do_single_iteration() |
||
32 |