From dda4dedca0459fc7c00eb1d9cb07e14af1621e0f Mon Sep 17 00:00:00 2001
From: Yanbo Liang <ybliang8@gmail.com>
Date: Thu, 19 Mar 2015 11:10:20 -0400
Subject: [PATCH] [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
---
.../spark/mllib/classification/LogisticRegression.scala | 4 ++++
.../scala/org/apache/spark/mllib/classification/SVM.scala | 4 ++++
.../mllib/regression/GeneralizedLinearAlgorithm.scala | 7 ++++++-
3 files changed, 14 insertions(+), 1 deletion(-)
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}"
+ }
}
/**
--
GitLab