首页 > 解决方案 > 在测试集上使用 XGBoost 的 AUC 实现

问题描述

我希望能够使用配备 XGBoost 的分类器来计算测试集上的 AUC(ROC 曲线下的面积),但使用 XGBoost 的 AUC 实现(用于计算训练时提供的测试集上的度量) . 所以,给定一个适合的模型

model = xgb.XGBClassifier(...)
model.fit(...)

我希望能够做类似的事情

model.evaluate(my_test_data, metric="auc")

这可能吗?

标签: xgboostauc

解决方案


推荐阅读