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.

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] Support custom metric functions #3956

Closed
jameslamb opened this issue Feb 14, 2021 · 3 comments
Closed

[dask] Support custom metric functions #3956

jameslamb opened this issue Feb 14, 2021 · 3 comments

Comments

@jameslamb
Copy link
Collaborator

Summary

The Dask estimators in lightgbm.dask should support the use of a custom objective function.

Motivation

Defining custom metrics functions gives users fine-grained ability to evaluate what's happening during training, and to control the training process via early stopping (#3712).

This feature would also bring Dask estimators closer to parity with the sklearn estimators.

Description

I haven't thought through this much yet, just writing up a placeholder issue for discussion. If you're reading this and have ideas, please comment and the issue can be re-opened.

References

See LINK for an explanation of how this works in the sklearn estimators.

Can look at how xgboost.dask handles this for inspiration:

At a quick glance it seems like xgboost.dask just passes the function all the way through to training and it "just works". Maybe we'll get lucky 😂

@jameslamb
Copy link
Collaborator Author

Closing this in favor of putting it in #2302 with other feature requests. Please comment if you're interested in working on this and the issue can be re-opened.

@jameslamb
Copy link
Collaborator Author

Re-opening this. I think it can be closed by a pull request that is currently in progress: #4101 (comment)

@jameslamb jameslamb reopened this Mar 30, 2021
@jameslamb
Copy link
Collaborator Author

fixed by #4101, thanks @ffineis !

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

No branches or pull requests

1 participant