aboutsummaryrefslogtreecommitdiff
path: root/mllib/src
diff options
context:
space:
mode:
authorYanbo Liang <ybliang8@gmail.com>2015-03-19 11:10:20 -0400
committerSean Owen <sowen@cloudera.com>2015-03-19 11:10:20 -0400
commitdda4dedca0459fc7c00eb1d9cb07e14af1621e0f (patch)
tree94e1df159c1ccecccfcbdfb50c498b8711607635 /mllib/src
parent3c4e486b9c8d3f256e801db7c176ab650c976135 (diff)
downloadspark-dda4dedca0459fc7c00eb1d9cb07e14af1621e0f.tar.gz
spark-dda4dedca0459fc7c00eb1d9cb07e14af1621e0f.tar.bz2
spark-dda4dedca0459fc7c00eb1d9cb07e14af1621e0f.zip
[SPARK-6291] [MLLIB] GLM toString & toDebugString
GLM toString prints out intercept, numFeatures. For LogisticRegression and SVM model, toString also prints out numClasses, threshold. GLM toDebugString prints out the whole weights, intercept. Author: Yanbo Liang <ybliang8@gmail.com> Closes #5038 from yanboliang/spark-6291 and squashes the following commits: 2f578b0 [Yanbo Liang] code format 78b33f2 [Yanbo Liang] fix typos 1e8a023 [Yanbo Liang] GLM toString & toDebugString
Diffstat (limited to 'mllib/src')
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala4
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala4
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/regression/GeneralizedLinearAlgorithm.scala7
3 files changed, 14 insertions, 1 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala b/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala
index b787667b01..e7c3599ff6 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/classification/LogisticRegression.scala
@@ -163,6 +163,10 @@ class LogisticRegressionModel (
}
override protected def formatVersion: String = "1.0"
+
+ override def toString: String = {
+ s"${super.toString}, numClasses = ${numClasses}, threshold = ${threshold.get}"
+ }
}
object LogisticRegressionModel extends Loader[LogisticRegressionModel] {
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala b/mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala
index cfc7f868a0..52fb62dcff 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala
@@ -86,6 +86,10 @@ class SVMModel (
}
override protected def formatVersion: String = "1.0"
+
+ override def toString: String = {
+ s"${super.toString}, numClasses = 2, threshold = ${threshold.get}"
+ }
}
object SVMModel extends Loader[SVMModel] {
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/regression/GeneralizedLinearAlgorithm.scala b/mllib/src/main/scala/org/apache/spark/mllib/regression/GeneralizedLinearAlgorithm.scala
index b262bec904..45b9ebb4cc 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/regression/GeneralizedLinearAlgorithm.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/regression/GeneralizedLinearAlgorithm.scala
@@ -76,7 +76,12 @@ abstract class GeneralizedLinearModel(val weights: Vector, val intercept: Double
predictPoint(testData, weights, intercept)
}
- override def toString() = "(weights=%s, intercept=%s)".format(weights, intercept)
+ /**
+ * Print a summary of the model.
+ */
+ override def toString: String = {
+ s"${this.getClass.getName}: intercept = ${intercept}, numFeatures = ${weights.size}"
+ }
}
/**