XGBoost在我的机器上运行缓慢,这是正常的吗?

时间:2018-01-10 20:51:16

标签: python data-science xgboost

我正在使用XGBoost来训练Kaggle项目Rossmann Store Sales的100万行和~15个功能。看起来很慢。在没有参数调整的情况下训练模型需要30分钟。如果我运行GridSearchCV训练模型有3倍和6个学习率值,则返回需要10个多小时。由于这是我第一次使用XGBoost,我不知道这是否正常。我无法想象调整XGBoost模型的所有参数需要多少天。 请帮帮我。

模型参数:XGBRegressor(learning_rate = 0.1,max_depth = 5,n_estimators = 1165,subsample = 0.8,colsample_bytree = 0.8,seed = 27)。我使用n_estimators 1165,因为它是xgboost.train返回的最佳迭代。同时将nthread从1更改为4,它根本不会改善性能。

我的电脑配置是; CPU:intel i7 6500U(2cores 4threads) 内存:8GB 操作系统:Windows 10

谢谢! 贾斯汀

0 个答案:

没有答案