Skip to content

[FEA] Add QuantileRegressor to cuML #6325

Open
@esnvidia

Description

@esnvidia

Is your feature request related to a problem? Please describe.
A clear and concise description of what the problem is. Ex. I wish I could use cuML to do [...]

QuantileRegressor is missing from cuML

Describe the solution you'd like
A clear and concise description of what you want to happen.

from cuml.linear_model import QuantileRegressor

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.
N/A

Additional context
Add any other context, code examples, or references to existing implementations about the feature request here.
QuantileRegression is important for obtaining probabilistic outputs.

Also, GPUs are beefy enough to fit multiple quantiles at once, so that would be a nice addition as well. something like

QuantileRegressor(quantile=[0.05, 0.5, 0.95]) would effectively fit all 3 models in parallel.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions