-
Notifications
You must be signed in to change notification settings - Fork 7
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
Conversation
Fix bug where quantile regression would throw exception when predicting on a single row.
Update regressor.py
Add test for single row quantile prediction.
There was a problem hiding this 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.
Add update as requested by crflynn to simplify reshape
Updated to remove unnecessary if statement.
Failure here is unrelated to your changes. I'll fix it and get it merged later today. |
Awesome! I did also add a unit test for this edge case so hopefully that will keep this issue from coming back as well. |
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.