Python XGBoost.set_params Beispiele

Beispiel #1

Datei: model_builder.py Projekt: glorysdj/analytics-zoo

    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

Beispiel #2

Datei: xgb_predictor.py Projekt: xiejinglei/analytics-zoo

 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