xgboost速度慢的原因是什么?

时间:2016-12-17 12:25:34

标签: python machine-learning anaconda data-mining xgboost

我在python中使用xgboost来解决多分类问题。输入训练数据的数量大约是500000.这是我的python代码执行速度很慢:

param = {'max_depth':23, 'eta':0.1, 'objective':'multi:softprob','num_class':24 }
num_round = 10
watchlist = [(dtrain, 'train')]    
bst = xgb.train(param, dtrain, num_round, watchlist)
preds = bst.predict(dtest)

我需要大约2个小时来进行训练并预测短语。然而,其他一些人'程序只需要大约5分钟来训练和预测。他们的代码与我的代码几乎完全相同。 速度极慢的原因是什么?我使用64位anaconda python 2.7。不同版本的python或xgboost是否有可能导致速度上的巨大差异?

1 个答案:

答案 0 :(得分:0)

这是因为我使用的xgboost版本。现在一切都很好。