XGBoost на GPU убивает ядро ​​(в Ubuntu)

Question

Ядро умирает каждый раз, когда я запускаю XGBoost на GPU с этим кодом:

params = {'tree_method': 'gpu_exact', 'max_depth': 100, 'learning_rate': 0.1} dtrain= xgb.DMatrix (X_train, y_train) xgb.train (params, dtrain, evals = [(dtrain, "train")])

Я получаю:

Kernel died - restarting

Answer 1

Я думаю, что главная проблема - max_depth = 100, xgboost строит деревья слой за слоем, поэтому полное дерево глубиной 100 будет слишком большим. Попробуйте использовать значения до 12, например.