test_optimizer_base.py 文件源码

def test_build_new_model_xgboost(self):
        xgb_model = XGBClassifier(max_depth=3)
        xgb_opt = Optimizer(xgb_model,[], lambda x: x)
        new_model = xgb_opt.build_new_model({'max_depth': 2})
        self.assertEqual(new_model.get_params()['max_depth'], 2)