Skip to content

MAINT: ndarrays for score (and move ndarray logic from partial_fit to fit)#246

Merged
TomAugspurger merged 4 commits intodask:masterfrom
stsievert:fit-ndarrays
Jun 29, 2018
Merged

MAINT: ndarrays for score (and move ndarray logic from partial_fit to fit)#246
TomAugspurger merged 4 commits intodask:masterfrom
stsievert:fit-ndarrays

Conversation

@stsievert
Copy link
Copy Markdown
Member

@stsievert stsievert commented Jun 28, 2018

What does this PR implement?
Two changes:

  1. Modify ParallelPostFit to accepts score ndarrays if passed by using sklearn's get_scorer.
  2. Move the logic in Fit immediately in Incremental.partial_fit on concrete inputs #236 to Incremental.fit (which works because partial_fit calls fit).

from ._partial import fit
from ._utils import copy_learned_attributes
from .metrics import get_scorer, check_scoring
from sklearn.metrics import get_scorer as sklearn_get_scorer
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Slight preference for just using import sklearn.metrics and then sklearn.metrics.get_scorer (we'll soon have more space to work with :)

@TomAugspurger TomAugspurger merged commit 7ae7bd6 into dask:master Jun 29, 2018
@TomAugspurger
Copy link
Copy Markdown
Member

Thanks @stsievert!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants