Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[dask] Early stopping #3952

Closed
wants to merge 35 commits into from
Closed

[dask] Early stopping #3952

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
35 commits
Select commit Hold shift + click to select a range
08430cb
es WiP, need to add eval_sample_weight and eval_group
ffineis Feb 9, 2021
2b75814
add weight, group to dask es. WiP.
ffineis Feb 11, 2021
77766e9
dask es reorg
ffineis Feb 13, 2021
86341fc
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
8d82646
Update tests/python_package_test/test_dask.py
ffineis Feb 16, 2021
69110b4
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
69b45dd
Update tests/python_package_test/test_dask.py
ffineis Feb 16, 2021
ee5f157
Update tests/python_package_test/test_dask.py
ffineis Feb 16, 2021
c96be2f
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
04040fc
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
cf720c9
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
c0d8ae6
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
5d19b50
Update tests/python_package_test/test_dask.py
ffineis Feb 16, 2021
4421736
Update tests/python_package_test/test_dask.py
ffineis Feb 16, 2021
65f98e5
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
6d14586
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
6737acd
Update python-package/lightgbm/dask.py
ffineis Feb 16, 2021
7099204
applying changes to eval_set PR WiP
ffineis Feb 16, 2021
18cd42c
dask support for eval_names, eval_metric, eval_stopping_rounds
ffineis Feb 17, 2021
1335591
add evals_result checks and other eval_set attribute-related test che…
ffineis Feb 18, 2021
c67cae5
resolve conflix
ffineis Feb 19, 2021
609d8a0
fix lint errors in test_dask.py
ffineis Feb 19, 2021
0102bc7
drop group_shape from _lgbmmodel_doc_fit.format for non-rankers, add …
ffineis Feb 19, 2021
b03715a
add eval_at to test_dask eval_set ranker tests
ffineis Feb 19, 2021
dfb3c72
add back group_shape to lgbmmmodel docs, tighten tests
ffineis Feb 19, 2021
82cf159
drop random eval weights from early stopping, probably causing traini…
ffineis Feb 20, 2021
8341074
add eval data templates to sklearn fit docs, add eval data docs to dask
ffineis Feb 20, 2021
d232b0b
add n_features to _create_data, eval_set tests stop w/ desirable tree…
ffineis Feb 20, 2021
e56251d
import alphabetically
ffineis Feb 20, 2021
10f6e58
resolve merge conflix
ffineis Feb 23, 2021
96fe750
fix conflix from 3994
ffineis Feb 24, 2021
b9cc723
add back get_worker for eval_set error handling
ffineis Feb 24, 2021
f593fd5
test_dask argmin typo
ffineis Feb 24, 2021
1548559
push forgotten eval_names bugfix
ffineis Feb 25, 2021
22f1ba4
eval_stopping_rounds -> early_stopping_rounds, fix failing non-es test
ffineis Feb 26, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion python-package/lightgbm/compat.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -114,13 +114,14 @@ def _check_sample_weight(sample_weight, X, dtype=None):
from dask.array import Array as dask_Array
from dask.dataframe import DataFrame as dask_DataFrame
from dask.dataframe import Series as dask_Series
from dask.distributed import Client, default_client, wait
from dask.distributed import Client, default_client, get_worker, wait
DASK_INSTALLED = True
except ImportError:
DASK_INSTALLED = False
delayed = None
Client = object
default_client = None
get_worker = None
wait = None

class dask_Array:
Expand Down
Loading