From d5ab42ceb94feb3645578bf459b104fb0b0d5682 Mon Sep 17 00:00:00 2001
From: Yuhao Yang <hhbyyh@gmail.com>
Date: Thu, 28 Apr 2016 19:52:09 +0100
Subject: [PATCH] [SPARK-14916][MLLIB] A more friendly tostring for FreqItemset
in mllib.fpm
## What changes were proposed in this pull request?
jira: https://issues.apache.org/jira/browse/SPARK-14916
FreqItemset as the result of FPGrowth should have a more friendly toString(), to help users and developers.
sample:
{a, b}: 5
{x, y, z}: 4
## How was this patch tested?
existing unit tests.
Author: Yuhao Yang <hhbyyh@gmail.com>
Closes #12698 from hhbyyh/freqtos.
---
.../src/main/scala/org/apache/spark/mllib/fpm/FPGrowth.scala | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/fpm/FPGrowth.scala b/mllib/src/main/scala/org/apache/spark/mllib/fpm/FPGrowth.scala
index 4f4996f3be..9166faa54d 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/fpm/FPGrowth.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/fpm/FPGrowth.scala
@@ -327,5 +327,9 @@ object FPGrowth {
def javaItems: java.util.List[Item] = {
items.toList.asJava
}
+
+ override def toString: String = {
+ s"${items.mkString("{", ",", "}")}: $freq"
+ }
}
}
--
GitLab