pycharm运行xgb.Booster.predict报错：Process finished with exit code -1073740791 (0xC0000409)

最新推荐文章于 2024-02-09 00:12:19 发布

myourdream2

最新推荐文章于 2024-02-09 00:12:19 发布

阅读量1.1w

点赞数

分类专栏： Machine Learning

本文链接：https://blog.csdn.net/myourdream2/article/details/86603300

版权

在PyCharm中使用xgboost训练模型时遇到进程退出代码-1073740791 (0xC0000409)的错误。该错误出现在调用`xgb.Booster.predict`时。经过排查，确定是由于GPU内存不足导致。解决方案是在训练时使用GPU，但在预测时切换到CPU进行预测，通过设置`tree_method`参数实现。问题解决后，代码运行正常。

摘要由CSDN通过智能技术生成

pycharm运行xgboost训练程序时，报错如下：

Process finished with exit code -1073740791 (0xC0000409)

调试发现是运行如下这行代码出现的问题：
dataset3_predict[‘label’] = model.predict(predict_dmatrix)
找遍了网络上各种方法都不行，但是网络上很多博客说这样的错误是显卡内存不足（如博客：https://blog.csdn.net/cuicheng01/article/details/79774156）

不断调试摸索，怀疑是xgb.Booster.predict使用GPU运行时显卡内存不足，因此考虑在使用GPU训练之后，在predict时，使用CPU predict，具体做法是：
1）xgboost train时使用的参数为：

params = {'booster': 'gbtree',
              'objective': 'binary:logistic',
              'eval_metric': 'auc',
              'gamma': 0.1,
              'min_child_weight': 1.1,
              'max_depth': 5,
              'lambda&