XGBoost CV函数如何确定使用哪个估计器或经过训练的模型?

2024-10-01 00:18:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在学习XGBoost,并且正在使用python(3.x)。我遇到了XGBoost cv函数。假设我有两个模型gbt1和gbt2,我使用XGBClassifier创建了这两个模型。现在,我希望使用XGBoost的CV方法进行交叉验证。我注意到我不需要指定我在这里尝试优化的模型。我只需要通过参数和数据矩阵。我这里的问题是XGBoost如何确定使用什么模型或估计器

cv_df = xgb.cv(params, DTrain, num_boost_round = 5, nfold=n_folds,
            early_stopping_rounds= early_stopping)

Tags: 数据方法函数模型参数矩阵交叉cv