This repository has been archived by the owner on Jul 24, 2024. It is now read-only.
BUG Return correct prediction dims with and without lambda #53
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Instead of trying for a one-liner with magic
squeeze
behavior, only squeeze if predicting for a single lambda value, and then only squeeze off the last axis (the lambda path). This way we'll get the correct return shape for all input shapes, including single rows.PR #51 fixed 1-row predictions for the case of single-lambda predictions with
LogitNet
, but didn't properly handle multi-lambda predictions, and didn't fix 1-row predictions forElasticNet
.Closes #52 , #30 , #25 .