tests/python_package_test/test_dask.py · d9a96c90cb479cef87047ba20517d97982b563eb · tianlh / LightGBM-DCU

[dask] Support pred_contrib in Dask predict() methods (fixes #3713) (#3774) · d9a96c90

James Lamb authored Jan 22, 2021



* adding pred_contrib support

* add tests

* linting

* remove raw_score

* add pred kwargs

* faster tests

* Apply suggestions from code review
Co-authored-by: Nikita Titov <nekit94-08@mail.ru>

* changes to tests

* Update tests/python_package_test/test_dask.py
Co-authored-by: Nikita Titov <nekit94-08@mail.ru>
Co-authored-by: Nikita Titov <nekit94-08@mail.ru>

d9a96c90

test_dask.py 19.6 KB

Replace test_dask.py