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