[SPARK-9834][MLLIB] implement weighted least squares via normal equation#8588
Closed
mengxr wants to merge 8 commits into
Closed
[SPARK-9834][MLLIB] implement weighted least squares via normal equation#8588mengxr wants to merge 8 commits into
mengxr wants to merge 8 commits into
Conversation
|
Test build #41977 has finished for PR 8588 at commit
|
|
Test build #41979 has finished for PR 8588 at commit
|
|
Test build #41994 has finished for PR 8588 at commit
|
Member
There was a problem hiding this comment.
Do we need standardizeLabel? I think without regularization, with/without standardization will not change the solution.
Contributor
Author
There was a problem hiding this comment.
We don't need it but I think it is useful to list the values explicitly here.
Contributor
|
LGTM, did not check low level implementation |
Contributor
|
jenkins test this please |
|
Test build #42145 has finished for PR 8588 at commit
|
Contributor
Author
|
Merged into master. I will make follow-up PRs to do the refactoring. |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
The goal of this PR is to have a weighted least squares implementation that takes the normal equation approach, and hence to be able to provide R-like summary statistics and support IRLS (used by GLMs). The tests match R's lm and glmnet.
There are couple TODOs that can be addressed in future PRs:
dsprtoBLASIt would be nice to have this merged first because it blocks couple other features.
@dbtsai