add_gradient_boosting_regressor_constr

gurobi_ml.sklearn.gradient_boosting_regressor.add_gradient_boosting_regressor_constr(gp_model, gradient_boosting_regressor, input_vars, output_vars=None, **kwargs)

Formulate gradient_boosting_regressor into gp_model.

The formulation predicts the values of output_vars using input_vars according to gradient_boosting_regressor. See our User’s Guide for details on the mip formulation used.

Parameters:

• gp_model (gurobipy model) – The gurobipy model where the predictor should be inserted.

• gradient_boosting_regressor (sklearn.ensemble.GradientBoostingRegressor) – The gradient boosting regressor to insert as predictor.

• input_vars (mvar_array_like) – Decision variables used as input for gradient boosting regressor in model.

• output_vars (mvar_array_like, optional) – Decision variables used as output for gradient boosting regressor in model.

Returns:

Object containing information about what was added to gp_model to formulate gradient_boosting_regressor.

Return type:

GradientBoostingRegressorConstr

Notes

See add_predictor_constr for acceptable values for input_vars and output_vars

Also see gurobi_ml.sklearn.decision_tree_regressor.add_decision_tree_regressor() for specific parameters to model decision tree estimators.