diff options
| author | y-shimizu <y.shimizu0429@gmail.com> | 2015-09-27 16:36:03 +0100 |
|---|---|---|
| committer | Sean Owen <sowen@cloudera.com> | 2015-09-27 16:36:03 +0100 |
| commit | 299b439920f980cce4c4f4e4a8436a5145efeaa3 (patch) | |
| tree | 1e73c189feb522860397f503a39fb0aa190d6f4f /mllib/src | |
| parent | 6f94d56a95e8c3a410a8d0c6a24ccca043227ba9 (diff) | |
| download | spark-299b439920f980cce4c4f4e4a8436a5145efeaa3.tar.gz spark-299b439920f980cce4c4f4e4a8436a5145efeaa3.tar.bz2 spark-299b439920f980cce4c4f4e4a8436a5145efeaa3.zip | |
[SPARK-10778] [MLLIB] Implement toString for AssociationRules.Rule
I implemented toString for AssociationRules.Rule, format like `[x, y] => {z}: 1.0`
Author: y-shimizu <y.shimizu0429@gmail.com>
Closes #8904 from y-shimizu/master.
Diffstat (limited to 'mllib/src')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala b/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala index 95c688c86a..07eb750b06 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala @@ -142,5 +142,10 @@ object AssociationRules { def javaConsequent: java.util.List[Item] = { consequent.toList.asJava } + + override def toString: String = { + s"${antecedent.mkString("{", ",", "}")} => " + + s"${consequent.mkString("{", ",", "}")}: ${confidence}" + } } } |