Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
RF: Add Gamma and Inverse Gaussian loss criteria (#4216)
This PR adds the Gamma and Inverse Gaussian Criteria to train decision trees, along with modifications to rf unit tests. --- checklist: - [x] Add Gamma and Inverse Gaussian Objective classes - [x] Add C++ tests for above - [x] Add remaining C++ tests for other objective functions: entropy and mean squared error - [x] Add python level convergence tests for gamma and inverse gaussian ( just like the one added for poison loss in #4156 ) - [x] Check for regressions by benchmarking on gbm-bench - [x] Convergence plots showing model trained on particular criteria performs better on it's own loss metric than a baseline (`mse`) Authors: - Venkat (https://github.com/venkywonka) Approvers: - Rory Mitchell (https://github.com/RAMitchell) - William Hicks (https://github.com/wphicks) - Dante Gama Dessavre (https://github.com/dantegd) URL: #4216
- Loading branch information