| 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 |