Repository navigation
[SPARK-6256] [MLlib] MLlib Python API parity check for regression - #4997
yanboliang wants to merge 3 commits into
Conversation
|
Test build #28509 has started for PR 4997 at commit
|
|
Test build #28509 has finished for PR 4997 at commit
|
|
Test PASSed. |
|
@jkbradley @mengxr Can you review this patch? |
|
@yanboliang |
|
Let's also add setIntercept. Also, in addition to setFeatureScaling being private, we do not need to expose optimizer.setUpdater for the 2 algorithms you listed because they have fixed updaters they should use (corresponding to the regularization they use). |
|
Test build #28978 has started for PR 4997 at commit
|
|
Test build #28978 has finished for PR 4997 at commit
|
|
Test PASSed. |
There was a problem hiding this comment.
Yes, "addIntercept" should be more clear and consistent.
|
Test build #29090 has started for PR 4997 at commit
|
|
Test build #29090 has finished for PR 4997 at commit
|
|
Test PASSed. |
There was a problem hiding this comment.
I'm sorry! I got confused about "intercept," thinking it was being added to this class. We should stick with the original name ("intercept") since it's a public API change otherwise.
|
@yanboliang It looks fine to me, except for the intercept issue (sorry!) and for doc tests. Could you please add doc tests for LassoWithSGD, RidgeRegressionWithSGD using setIntercept, setValidateData? Thanks! |
|
Test build #29160 has started for PR 4997 at commit
|
|
Test build #29160 has finished for PR 4997 at commit
|
|
Test PASSed. |
|
@yanboliang LGTM merging into master |
MLlib Python API parity check for Regression, major disparities need to be added for Python list following:
setFeatureScaling is mllib private function which is not needed to expose in pyspark.