GBDT与XGBoost速度对比

Posted dulingtingzi

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了GBDT与XGBoost速度对比相关的知识,希望对你有一定的参考价值。

同样的数据量,在同样的环境下的速度对比

正负样本数量:142287,711435

参数设置:

model_xgb=XGBClassifier(learning_rate=0.1,max_depth=5,n_estimators=300,subsample=0.6,objective='binary:logistic',n_jobs=10,enable_categorical=True)
#model_GBDT = GradientBoostingClassifier(max_depth=5, n_estimators=300, subsample=0.6, max_features=0.5)

用时对比:

GBDT:

predict used time: 2145.752547979355
train and predict used time all: 10983.663108110428

XGBoost:

predict used time: 1919.792335987091
train and predict used time all: 9046.629899263382
 

以上是关于GBDT与XGBoost速度对比的主要内容,如果未能解决你的问题,请参考以下文章

GBDT与XGBoost速度对比

GBDT与XGBoost速度对比

XGBoost与GBDT(一)-几种最优化方法对比

GBDT与XGBoost

LightGBM算法 & XGBoost算法对比分析

XGBoost算法