This is the implementation of paper Task-wise Split Gradient Boosting Trees on XGBoost. install the python package of xgboost.
make
cd python-package
sudo python setup.py install
About how to use TSGB in python, see details here
If you find this repository helpful in your research, kindly consider citing the paper mentioned below:
@inproceedings{chen2021task,
title={Task-wise split gradient boosting trees for multi-center diabetes prediction},
author={Chen, Mingcheng and Wang, Zhenghui and Zhao, Zhiyun and Zhang, Weinan and Guo, Xiawei and Shen, Jian and Qu, Yanru and Lu, Jieli and Xu, Min and Xu, Yu and others},
booktitle={Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery \& Data Mining},
pages={2663--2673},
year={2021}