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 в системе.
Обсуждают сегодня