预测时Xgboost(GPU)崩溃 [英] Xgboost (GPU) crashing while predicting

问题描述

我在Python中使用XGBoost GPU版本，每当我尝试运行.predict时，它就会崩溃.它适用于较小的数据集，但对于我当前的问题，它不起作用.

I am using XGBoost GPU version in Python and it crashes whenever I try to run .predict. It works for a smaller data set but for my current problem it is not working.

train_final.shape, test_final.shape
((631761, 174), (421175, 174))


params = {
          'objective': 'multi:softmax', 
          'eval_metric': 'mlogloss',
          'eta': 0.1,
          'max_depth': 6, 
          'nthread': 4,
          'alpha':0,
          'num_class': 5,
          'random_state': 42, 
          'tree_method': 'gpu_hist',
          'silent': True
             }

GPU统计:GTX 1070，6 GB 内存:32 GB

GPU Stats: GTX 1070, 6 GB RAM: 32 GB

有人可以帮我理解为什么会这样吗?

Could someone please help me understand why is this happening?

推荐答案

保存该模型，先删除增强器，然后再次加载模型即可.

# training
bst = xgb.train(param, dtrain, num_round)

#save model
joblib.dump(bst, 'xgb_model.dat')
bst.__del__()

#load saved model
bst = joblib.load('xgb_model.dat')
preds = bst.predict(dtest)

这篇关于预测时Xgboost(GPU)崩溃的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！