We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Feature Interactions in XGBoost

Abstract: In this paper, we investigate how feature interactions can be identified to be used as constraints in the gradient boosting tree models using XGBoost's implementation. Our results show that accurate identification of these constraints can help improve the performance of baseline XGBoost model significantly. Further, the improvement in the model structure can also lead to better interpretability.
Comments: 7 pages, 2 Figures
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Cite as: arXiv:2007.05758 [cs.LG]
  (or arXiv:2007.05758v1 [cs.LG] for this version)

Submission history

From: Kshitij Goyal [view email]
[v1] Sat, 11 Jul 2020 12:17:40 GMT (2161kb)

Link back to: arXiv, form interface, contact.