XGBoost 中的特征交互
机器学习
2020-07-14 v1 机器学习
摘要
本文研究如何识别特征交互，并将其作为梯度提升树模型中使用 XGBoost 实现时的约束。我们的结果表明，准确识别这些约束可显著改善基线 XGBoost 模型的性能。此外，模型结构的改进亦可带来更好的可解释性。
引用
@article{arxiv.2007.05758,
title = {Feature Interactions in XGBoost},
author = {Kshitij Goyal and Sebastijan Dumancic and Hendrik Blockeel},
journal= {arXiv preprint arXiv:2007.05758},
year = {2020}
}
备注
7 pages, 2 Figures