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

Prepare external memory support for hist. #7638

Merged
merged 5 commits into from Feb 10, 2022

Conversation

trivialfis
Copy link
Member

Extracted from #7531 .

This PR prepares the GHistIndexMatrix to host the column matrix which is used by the hist tree method by accepting sparse_threshold parameter.

Some cleanups are made to ensure the correct batch param is being passed into DMatrix along with some additional tests for correctness of SimpleDMatrix.

src/data/gradient_index.h Outdated Show resolved Hide resolved
@trivialfis trivialfis merged commit 2775c2a into dmlc:master Feb 10, 2022
@trivialfis trivialfis deleted the regen-pipeline branch February 10, 2022 08:58
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.

None yet

2 participants