
上QQ阅读APP看书,第一时间看更新
TensorFlow boosted trees estimator
TensorFlow Boosted Trees (TFBT) is an improved scalable ensemble model built on top of generic gradient boosting trees.
Google published the details of the TensorFlow boosted trees implementation in the following paper: A scalable TensorFlow based framework for gradient boosting by Natalia Ponomareva, Soroush Radpour, Gilbert Hendry, Salem Haykal, Thomas Colthurst, Petr Mitrichev, Alexander Grushetsky, presented at the European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases (ECML PKDD) 2017 . The paper is available at the following link: http://ecmlpkdd2017.ijs.si/papers/paperID705.pdf.
The gradient boosting algorithm is implemented by various libraries such as sklearn, MLLib, and XGBoost. TensorFlow's implementation is different from these implementations as described in the following table extracted from the TFBT research paper:
TFBT Research Paper from Google
The TFBT model can be extended by writing custom loss functions in TensorFlow. The differentiation for these custom loss functions is automatically provided by TensorFlow.