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

Support gamma in GPU_Hist. #4874

Merged
merged 2 commits into from
Sep 24, 2019
Merged

Support gamma in GPU_Hist. #4874

merged 2 commits into from
Sep 24, 2019

Conversation

trivialfis
Copy link
Member

@hcho3 @RAMitchell I just prevent the split when loss_chg < gamma. Could you think of a case adding an explicit pruner is useful? One additional thing I found in pruner is a syncing operation (broadcasting rank 0 tree), but I don't think it's necessary for gpu_hist.

@trivialfis
Copy link
Member Author

I will add tests if the method is reasonable.

@RAMitchell
Copy link
Member

I can't think of any reason why not, let's do it.

@trivialfis
Copy link
Member Author

@RAMitchell Please take a look again.

@trivialfis trivialfis merged commit 0b89cd1 into dmlc:master Sep 24, 2019
@trivialfis trivialfis deleted the gamma branch September 24, 2019 02:16
@trivialfis
Copy link
Member Author

@panranyang Fixed. #4707

@lock lock bot locked as resolved and limited conversation to collaborators Dec 23, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants