使用XGBoost的Learning API的scale_pos_weight

Question

我发现可以通过scale_pos_weight为XGBoost的Scikit-Learn API中的不平衡问题添加权重。它在Learning API中是否具有等效项？如果没有，

• 这背后有原因吗？
• 可以通过学习API以某种方式实现此校正因子/权重吗？

Answer 1

XGBoost学习API具有scale_pos_weight的参数，您可以在train函数的params字典中提供它。

您可以做这样的事情-

创建参数字典，例如-params = {'scale_pos_weight'：1，'max_depth'：3}

现在，您可以在学习API中使用它，例如-

model = xgb.train(params, dtrain, num_boost_round)