【AI实战】XGBRegressor模型加速训练,使用GPU秒级训练XGBRegressor
于 2022-06-07 11:31:38 首次发布
本文介绍了如何利用XGBRegressor模型,并结合GPU进行训练,以大幅缩短训练时间。在2百万级别的数据上,通过设置tree_method='gpu_hist'和gpu_id=3,模型在34.344秒内完成了训练。这展示了在大规模数据下,GPU加速对于机器学习模型训练的高效性。

最低0.47元/天 解锁文章
2505

被折叠的 条评论
为什么被折叠?



