Skip to content

Addition of Piecewise Quantile Loss#3054

Open
tatron wants to merge 1 commit intocatboost:masterfrom
wayfair-contribs:piecewise_quantile_branch
Open

Addition of Piecewise Quantile Loss#3054
tatron wants to merge 1 commit intocatboost:masterfrom
wayfair-contribs:piecewise_quantile_branch

Conversation

@tatron
Copy link

@tatron tatron commented Mar 5, 2026

This PR adds a new loss function called PiecewiseQuantile and closes #3053 . It uses different quantile values based on where a sample’s target/ground truth value falls within user-provided intervals.

As an example:
PiecewiseQuantile:boundaries=0,1;quantiles=0.25,0.5,0.75
This assigns:

  • alpha=0.25 when y <= 0
  • alpha=0.5 when 0 < y <= 1
  • alpha=0.75 when y > 1.

The length of these input lists are not fixed. To be clear, these quantiles are with respect to the other samples in the respective intervals.

We include the loss function, its gradients, and the corresponding metric. We also include unit tests for all of our additions.

@tatron
Copy link
Author

tatron commented Mar 9, 2026

I noticed that one of the checks failed, but this seems to be due to a Github internal server issue at the time of the CI run, rather than our code. I can confirm that when I built the wheel and ran tests locally on my MacOS, it passed without issue.

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.

Feature Contribution: Piecewise Quantile Loss

1 participant