Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • shiyu1994's avatar
    Clear split info buffer in cost efficient gradient boosting before every... · f1328d5c
    shiyu1994 authored 
    Clear split info buffer in cost efficient gradient boosting before every iteration (fix partially #3679) (#5164)

* clear split info buffer in cegb_ before every iteration

* check nullable of cegb_ in serial_tree_learner.cpp

* add a test case for checking the split buffer in CEGB

* swith to Threading::For instead of raw OpenMP

* apply review suggestions

* apply review comments

* remove device cpu
    f1328d5c
cost_effective_gradient_boosting.hpp 6.29 KB