update

Author: aoiang

branincurrin/Classifier.py

@@ -744,7 +744,7 @@ def optimize_qparego_and_get_observation(model, train_obj, sampler, path):
             candidates, _ = optimize_acqf_list(
                 acq_function_list=acq_func_list,
                 bounds=standard_bounds,
-                num_restarts=5,
+                num_restarts=20,
                 raw_samples=1024,
                 options={"batch_limit": 1, "maxiter": 2},
                 lamcts_boundry=path,

vehiclesafety/Classifier.py

@@ -723,7 +723,7 @@ def optimize_qparego_and_get_observation(model, train_obj, sampler, path):
             candidates, _ = optimize_acqf_list(
                 acq_function_list=acq_func_list,
                 bounds=standard_bounds,
-                num_restarts=5,
+                num_restarts=20,
                 raw_samples=1024,
                 options={"batch_limit": 1, "maxiter": 2},
                 lamcts_boundry=path,

vehiclesafety/MCTS.py

@@ -539,13 +539,13 @@ def search(self):
     parser.add_argument('--problem', type=str, default='bc', help='choose the problem')
     parser.add_argument('--data_id', type=int, default=-1, help='specific run id')
     parser.add_argument('--obj_id', type=int, default=-1, help='objective to split')
-    parser.add_argument('--kernel', type=str, default='poly', help='kernel type of svm')
+    parser.add_argument('--kernel', type=str, default='rbf', help='kernel type of svm')
     parser.add_argument('--gamma', type=str, default='scale', help='auto or scale')
     parser.add_argument('--degree', type=int, default=4, help='svm degree')
     parser.add_argument('--iter', type=int, default=18, help='total iterations')
     parser.add_argument('--sample_num', type=int, default=5, help='sample numsbers per iteration')
     parser.add_argument('--runs', type=int, default=5, help='total runs')
-    parser.add_argument('--cp', type=float, default=15, help='cp value in MCTS')
+    parser.add_argument('--cp', type=float, default=18, help='cp value in MCTS')
     parser.add_argument('--sample_method', type=str, default='bayesian', help='bayesian, cmaes or random')
     parser.add_argument('--cmaes_method', type=str, default='lamcts', help='lamcts or vanilla')
     parser.add_argument('--split_method', type=str, default='dominance', help='dominance or regressor')