8 Commits

Author	SHA1	Message	Date
Jiaming Yuan	142bdc73ec	[EM] Support SHAP contribution with QDM. (#10724) ... - Add GPU support. - Add external memory support. - Update the GPU tree shap.	2024-08-22 05:25:10 +08:00
Jiaming Yuan	8c676c889d	Remove internal use of gpu_id. (#9568)	2023-09-20 23:29:51 +08:00
Jiaming Yuan	972730cde0	Use matrix for gradient. (#9508) ... - Use the `linalg::Matrix` for storing gradients. - New API for the custom objective. - Custom objective for multi-class/multi-target is now required to return the correct shape. - Custom objective for Python can accept arrays with any strides. (row-major, column-major)	2023-08-24 05:29:52 +08:00
Rong Ou	bde1ebc209	Switch back to the GPUIDX macro (#9438)	2023-08-04 15:14:31 +08:00
Rong Ou	c2b85ab68a	Clean up MGPU C++ tests (#9430)	2023-08-02 14:31:18 +08:00
Jiaming Yuan	e206b899ef	Rework MAP and Pairwise for LTR. (#9075)	2023-04-28 02:39:12 +08:00
Jiaming Yuan	ef13dd31b1	Rework the NDCG objective. (#9015)	2023-04-18 21:16:06 +08:00
Jiaming Yuan	d062a9e009	Define pair generation strategies for LTR. (#8984)	2023-03-30 12:00:35 +08:00