2016-12-17 694 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版本。現在一切都很好。

相關問題