You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@spark.apache.org by db...@apache.org on 2015/07/21 07:15:17 UTC
spark git commit: [SPARK-9204][ML] Add default params test for
linearyregression suite
Repository: spark
Updated Branches:
refs/heads/master a3c7a3ce3 -> 4d97be953
[SPARK-9204][ML] Add default params test for linearyregression suite
Author: Holden Karau <ho...@pigscanfly.ca>
Closes #7553 from holdenk/SPARK-9204-add-default-params-test-to-linear-regression and squashes the following commits:
630ba19 [Holden Karau] style fix
faa08a3 [Holden Karau] Add default params test for linearyregression suite
Project: http://git-wip-us.apache.org/repos/asf/spark/repo
Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/4d97be95
Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/4d97be95
Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/4d97be95
Branch: refs/heads/master
Commit: 4d97be95300f729391c17b4c162e3c7fba09b8bf
Parents: a3c7a3c
Author: Holden Karau <ho...@pigscanfly.ca>
Authored: Mon Jul 20 22:15:10 2015 -0700
Committer: DB Tsai <db...@netflix.com>
Committed: Mon Jul 20 22:15:10 2015 -0700
----------------------------------------------------------------------
.../ml/regression/LinearRegressionSuite.scala | 25 ++++++++++++++++++++
1 file changed, 25 insertions(+)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/spark/blob/4d97be95/mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala
----------------------------------------------------------------------
diff --git a/mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala
index 374002c..7cdda3d 100644
--- a/mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala
+++ b/mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala
@@ -18,6 +18,7 @@
package org.apache.spark.ml.regression
import org.apache.spark.SparkFunSuite
+import org.apache.spark.ml.param.ParamsSuite
import org.apache.spark.mllib.linalg.{DenseVector, Vectors}
import org.apache.spark.mllib.util.{LinearDataGenerator, MLlibTestSparkContext}
import org.apache.spark.mllib.util.TestingUtils._
@@ -55,6 +56,30 @@ class LinearRegressionSuite extends SparkFunSuite with MLlibTestSparkContext {
}
+ test("params") {
+ ParamsSuite.checkParams(new LinearRegression)
+ val model = new LinearRegressionModel("linearReg", Vectors.dense(0.0), 0.0)
+ ParamsSuite.checkParams(model)
+ }
+
+ test("linear regression: default params") {
+ val lir = new LinearRegression
+ assert(lir.getLabelCol === "label")
+ assert(lir.getFeaturesCol === "features")
+ assert(lir.getPredictionCol === "prediction")
+ assert(lir.getRegParam === 0.0)
+ assert(lir.getElasticNetParam === 0.0)
+ assert(lir.getFitIntercept)
+ val model = lir.fit(dataset)
+ model.transform(dataset)
+ .select("label", "prediction")
+ .collect()
+ assert(model.getFeaturesCol === "features")
+ assert(model.getPredictionCol === "prediction")
+ assert(model.intercept !== 0.0)
+ assert(model.hasParent)
+ }
+
test("linear regression with intercept without regularization") {
val trainer = new LinearRegression
val model = trainer.fit(dataset)
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@spark.apache.org
For additional commands, e-mail: commits-help@spark.apache.org