Skip to content
This repository has been archived by the owner on Nov 27, 2023. It is now read-only.

update dask-ml to v1.0.0 #6

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

update dask-ml to v1.0.0 #6

wants to merge 2 commits into from

Conversation

zexuan-zhou
Copy link

No description provided.

@@ -120,7 +134,68 @@ def build_graph(
cache_cv=True,
multimetric=False,
):
# This is provided for compatibility with TPOT. Remove
# once TPOT is updated and requires a dask-ml>=0.13.0
def decompress_params(fields, params):
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This corresponds to gathering cv results as they finished:
dask#433

@@ -1219,21 +1267,59 @@ def fit(self, X, y=None, groups=None, **fit_params):
if scheduler is dask.threaded.get and n_jobs == 1:
scheduler = dask.local.get_sync

out = scheduler(dsk, keys, num_workers=n_jobs)
if "Client" in type(getattr(scheduler, "__self__", None)).__name__:
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

feature_union,
feature_union_concat,
fit,
fit_and_score,
fit_best,
fit_transform,
get_best_params,
pipeline,
score,
_get_fold_sample_weights,
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ryan's fix is there and unchanged

feature_union,
feature_union_concat,
fit,
fit_and_score,
fit_best,
fit_transform,
get_best_params,
pipeline,
score,
_get_fold_sample_weights,
get_sample_weights
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ryan's fix is there and unchanged

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants