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

Predict with QuantileDMatrix #8224

Open
ldesreumaux opened this issue Sep 5, 2022 · 1 comment
Open

Predict with QuantileDMatrix #8224

ldesreumaux opened this issue Sep 5, 2022 · 1 comment

Comments

@ldesreumaux
Copy link
Contributor

With the current codebase, it is not possible to use Learner::Predict with a QuantileDMatrix to get leaf indices, feature contributions or feature interactions. The following error is raised:

BatchSet<SparsePage> GetRowBatches() override {
LOG(FATAL) << "Not implemented.";
return BatchSet<SparsePage>(BatchIterator<SparsePage>(nullptr));
}

Is there any plan to support these prediction types?

@trivialfis
Copy link
Member

Yes, we are still working on this. For now, my priority is to make sure it can work with both CPU and GPU data: #8215 .

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

2 participants