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

Bugfix for quantile regression #114

Merged
merged 8 commits into from
Nov 27, 2021
Merged

Bugfix for quantile regression #114

merged 8 commits into from
Nov 27, 2021

Conversation

corwinjoy
Copy link
Contributor

Thanks for writing this great wrapper!
I'd like to propose a minor bugfix. It turns out that quantile regression throws an exception when it tries to predict on one row. I have added a reshape call to fix this edge case.

Fix bug where quantile regression would throw exception when predicting on a single row.
Add test for single row quantile prediction.
Copy link
Owner

@crflynn crflynn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for catching this. I just requested one small change. Otherwise looks good.

skranger/ensemble/regressor.py Outdated Show resolved Hide resolved
Add update as requested by crflynn to simplify reshape
Updated to remove unnecessary if statement.
@crflynn
Copy link
Owner

crflynn commented Nov 22, 2021

Failure here is unrelated to your changes. I'll fix it and get it merged later today.

@corwinjoy
Copy link
Contributor Author

Awesome! I did also add a unit test for this edge case so hopefully that will keep this issue from coming back as well.

@crflynn crflynn merged commit a15c67b into crflynn:master Nov 27, 2021
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