optuna

1
import json
2
import sys
3

4
from kurobako import solver
5
from kurobako.solver.optuna import OptunaSolverFactory
6
import optuna
7

8

9
optuna.logging.disable_default_handler()
10

11

12
def create_study(seed: int) -> optuna.Study:
13
    # Avoid the fail by `flake8`.
14
    seed
15

16
    n_objectives = 2
17
    directions = ["minimize"] * n_objectives
18

19
    sampler_name = sys.argv[1]
20

21
    # Sampler.
22
    sampler_cls = getattr(
23
        optuna.samplers,
24
        sampler_name,
25
        getattr(optuna.integration, sampler_name, None),
26
    )
27
    if sampler_cls is None:
28
        raise ValueError("Unknown sampler: {}.".format(sampler_name))
29

30
    sampler_kwargs = json.loads(sys.argv[2])
31
    sampler = sampler_cls(**sampler_kwargs)
32

33
    return optuna.create_study(
34
        directions=directions,
35
        sampler=sampler,
36
        pruner=optuna.pruners.NopPruner(),
37
    )
38

39

40
if __name__ == "__main__":
41
    factory = OptunaSolverFactory(create_study)
42
    runner = solver.SolverRunner(factory)
43
    runner.run()
44