dtlz2/configurations/dtlz2_ax_joblib.json

0001 {
0002   "problem": {
0003     "name": "DTLZ2 Ax with JobLib",
0004     "problem_type": "DTLZ2",
0005     "output_location": "./output/ax_joblib",
0006     "work_location": "./work/ax_joblib",
0007     "design_parameters_file": "../dtlz2_design.params",
0008     "objectives": [
0009       {
0010         "name": "f1",
0011         "direction": "minimize"
0012       },
0013       {
0014         "name": "f2",
0015         "direction": "minimize"
0016       }
0017     ]
0018   },
0019   "optimizer": {
0020     "name": "ax",
0021     "type": "bayesian",
0022     "parameters": {
0023       "initialization_strategy": "sobol",
0024       "generator": "BOTORCH_MODULAR",
0025       "generator_kwargs": {
0026         "surrogate_spec": {
0027           "model_configs": [
0028             {
0029               "botorch_model_class": "SingleTaskGP"
0030             }
0031           ]
0032         },
0033         "botorch_acqf_class": "qLogNoisyExpectedHypervolumeImprovement",
0034         "acquisition_options": {
0035           "prune_baseline": true
0036         }
0037       },
0038       "generator_gen_kwargs": {
0039         "model_gen_options": {
0040           "optimizer_kwargs": {
0041             "sequential": false,
0042             "num_restarts": 10
0043           }
0044         }
0045       },
0046       "objective_thresholds": {
0047         "f1": 1.0,
0048         "f2": 1.0
0049       },
0050       "n_initial_samples": 20,
0051       "n_iterations": 100,
0052       "batch_size": 4,
0053       "seed": 42
0054     }
0055   },
0056   "scheduler": {
0057     "runner_type": "JobLibRunner",
0058     "parameters": {
0059       "n_jobs": 4,
0060       "backend": "threading"
0061     }
0062   },
0063   "workflows": {
0064     "workflows": [
0065       {
0066         "name": "dtlz2_eval",
0067         "description": "DTLZ2 objective evaluation",
0068         "branches": [
0069           {
0070             "name": "main",
0071             "stages": [
0072               {
0073                 "name": "evaluate",
0074                 "objective_plan": {
0075                   "steps": {
0076                     "stages": [
0077                       {
0078                         "name": "evaluate",
0079                         "inline": {
0080                           "entrypoint": "examples.dtlz2.dtlz2_evaluator:objective_payload"
0081                         },
0082                         "produces_objective": true
0083                       }
0084                     ]
0085                   }
0086                 }
0087               }
0088             ]
0089           }
0090         ]
0091       }
0092     ]
0093   }
0094 }