Coverage for jetgp/hyperparameter_optimizers/jade.py: 100%

1from jetgp.utils import jade as core_jade 

2 

3def jade(func, lb, ub, **kwargs): 

4 """ 

5 Wrapper for JADE with unified interface. 

6 """ 

7 pop_size = kwargs.pop("pop_size", 20) 

8 n_generations = kwargs.pop("n_generations", 50) 

9 local_opt_every = kwargs.pop("local_opt_every", 15) 

10 initial_positions = kwargs.pop("initial_positions", None) 

11 p = kwargs.pop("p", 0.1) 

12 c = kwargs.pop("c", 0.1) 

13 seed = kwargs.pop("seed", 42) 

14 debug = kwargs.pop("debug", False) 

15 

16 for key in ["n_restart_optimizer", "optimizer", "debug", "x0"]: 

17 kwargs.pop(key, None) 

18 

19 return core_jade( 

20 func, 

21 lb, 

22 ub, 

23 pop_size=pop_size, 

24 n_generations=n_generations, 

25 p=p, 

26 c=c, 

27 local_opt_every=local_opt_every, 

28 initial_positions=initial_positions, 

29 seed=seed, 

30 debug=debug, 

31 **kwargs 

32 )