2016-12-17 696 views
-1

我在python中使用xgboost来解决多分类问题。输入训练数据的数量约为50万。这是执行很慢我的Python代码:
xgboost速度慢的原因是什么?

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在速度上造成巨大差异?

回答

0

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

相关问题