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速度对比的主要内容,如果未能解决你的问题,请参考以下文章