def build(self, config): from zoo.orca.automl.xgboost.XGBoost import XGBoost model = XGBoost(model_type=self.model_type, config=config) if self.n_cpus is not None: model.set_params(n_jobs=self.n_cpus) return model
def create_model(): _model = XGBoost(model_type=model_type, config=config) if "cpu" in resources_per_trial: _model.set_params(n_jobs=resources_per_trial.get("cpu")) return _model