Passed
Pull Request — master (#110)
by
unknown
01:23
created

CustomOptimizationStrategy.add_optimizer()   A

Complexity

Conditions 1

Size

Total Lines 8
Code Lines 7

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 7
nop 4
dl 0
loc 8
rs 10
c 0
b 0
f 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