diff options
author | lewuathe <lewuathe@me.com> | 2015-10-19 10:46:10 -0700 |
---|---|---|
committer | DB Tsai <dbt@netflix.com> | 2015-10-19 10:46:10 -0700 |
commit | 4c33a34ba3167ae67fdb4978ea2166ce65638fb9 (patch) | |
tree | 5a5dbae89a230ad0c82acab25ba98e9121b1af6b /mllib/src/test/java/org | |
parent | dfa41e63b98c28b087c56f94658b5e99e8a7758c (diff) | |
download | spark-4c33a34ba3167ae67fdb4978ea2166ce65638fb9.tar.gz spark-4c33a34ba3167ae67fdb4978ea2166ce65638fb9.tar.bz2 spark-4c33a34ba3167ae67fdb4978ea2166ce65638fb9.zip |
[SPARK-10668] [ML] Use WeightedLeastSquares in LinearRegression with L…
…2 regularization if the number of features is small
Author: lewuathe <lewuathe@me.com>
Author: Lewuathe <sasaki@treasure-data.com>
Author: Kai Sasaki <sasaki@treasure-data.com>
Author: Lewuathe <lewuathe@me.com>
Closes #8884 from Lewuathe/SPARK-10668.
Diffstat (limited to 'mllib/src/test/java/org')
-rw-r--r-- | mllib/src/test/java/org/apache/spark/ml/regression/JavaLinearRegressionSuite.java | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mllib/src/test/java/org/apache/spark/ml/regression/JavaLinearRegressionSuite.java b/mllib/src/test/java/org/apache/spark/ml/regression/JavaLinearRegressionSuite.java index 91c589d00a..4fb0b0d109 100644 --- a/mllib/src/test/java/org/apache/spark/ml/regression/JavaLinearRegressionSuite.java +++ b/mllib/src/test/java/org/apache/spark/ml/regression/JavaLinearRegressionSuite.java @@ -61,6 +61,7 @@ public class JavaLinearRegressionSuite implements Serializable { public void linearRegressionDefaultParams() { LinearRegression lr = new LinearRegression(); assertEquals("label", lr.getLabelCol()); + assertEquals("auto", lr.getSolver()); LinearRegressionModel model = lr.fit(dataset); model.transform(dataset).registerTempTable("prediction"); DataFrame predictions = jsql.sql("SELECT label, prediction FROM prediction"); @@ -75,7 +76,7 @@ public class JavaLinearRegressionSuite implements Serializable { // Set params, train, and check as many params as we can. LinearRegression lr = new LinearRegression() .setMaxIter(10) - .setRegParam(1.0); + .setRegParam(1.0).setSolver("l-bfgs"); LinearRegressionModel model = lr.fit(dataset); LinearRegression parent = (LinearRegression) model.parent(); assertEquals(10, parent.getMaxIter()); |