-
Notifications
You must be signed in to change notification settings - Fork 29.1k
[SPARK-18434][ML][FOLLOWUP] Add checking for setSolver in GLM #15955
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
Conversation
|
|
|
Test build #68918 has finished for PR 15955 at commit
|
|
@zhengruifeng It's better we can define a variable |
943edba to
1fe2f92
Compare
|
@yanboliang I updated this pr according to your comments. And now it's |
|
Test build #68977 has finished for PR 15955 at commit
|
yanboliang
left a comment
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.
One minor comment, otherwise, LGTM. Thanks.
| override def load(path: String): GeneralizedLinearRegression = super.load(path) | ||
|
|
||
| /** String name for "irls" solver. */ | ||
| private[regression] val IRLS = "irls" |
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.
Could we make it more private (i.e. private[GeneralizedLinearRegression])?
| object LinearRegression extends DefaultParamsReadable[LinearRegression] { | ||
|
|
||
| /** String name for "auto" solver. */ | ||
| private[regression] val AUTO = "auto" |
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.
Ditto.
|
@yanboliang I add the |
|
After offline discussion with @yanboliang , I will update this PR to make solvers more private and include MLPC in this PR. |
|
After offline discussion with @zhengruifeng , we think the validation in the |
What changes were proposed in this pull request?
add checking in GLM.setSolver to forbidden unsupported solvers:
val glm = new GeneralizedLinearRegression().setSolver("123")How was this patch tested?
existing tests
cc @yanboliang