params = {
'objective': 'reg:linear',
'booster':'gblinear',
'gpu_id': 0,
'tree_method': 'gpu_hist'
}
bst = xgb.train(params ...
Делаю так, быстрее не становится.. Тот же catboost быстрее раз в 10 на этом же датасете.. Или Catboost на самом деле гораздо быстрее XGBoost?
Вообще, Catboost действительно быстрее xgboost'а. Попробуйте убрать gpu_id, если у вас одна GPU.
Не эксперт но catboost и lightgbm вроде как должны быть быстрее
Блин, убрал этот ноль и точно заработало.. Либо колаб тупит, сразу не включил gpu.. XGBoost по ощущениям даже быстрее катбуста, не замерял конечно))
Или сложности с нумерацией GPU в системе.
Обсуждают сегодня