SimonBlanke /
Hyperactive
| Total Complexity | 2 |
| Total Lines | 23 |
| Duplicated Lines | 0 % |
| Changes | 0 | ||
| 1 | # Author: Simon Blanke |
||
| 2 | # Email: [email protected] |
||
| 3 | # License: MIT License |
||
| 4 | |||
| 5 | from .optimization_strategy import BaseOptimizationStrategy |
||
| 6 | |||
| 7 | |||
| 8 | class CustomOptimizationStrategy(BaseOptimizationStrategy): |
||
| 9 | def __init__(self): |
||
| 10 | super().__init__() |
||
| 11 | |||
| 12 | self.optimizer_setup_l = [] |
||
| 13 | self.duration_sum = 0 |
||
| 14 | |||
| 15 | def add_optimizer(self, optimizer, duration=1, early_stopping=None): |
||
| 16 | self.duration_sum += duration |
||
| 17 | optimizer_setup = { |
||
| 18 | "optimizer": optimizer, |
||
| 19 | "duration": duration, |
||
| 20 | "early_stopping": early_stopping, |
||
| 21 | } |
||
| 22 | self.optimizer_setup_l.append(optimizer_setup) |
||
| 23 |
