Github user WeichenXu123 commented on a diff in the pull request:
https://github.com/apache/spark/pull/13493#discussion_r141852693
--- Diff: python/pyspark/mllib/tests.py ---
@@ -1762,6 +1763,18 @@ def test_pca(self):
self.assertEqualUpToSign(pcs.toArray()[:, k - 1],
expected_pcs[:, k - 1])
+class FPGrowthTest(MLlibTestCase):
+
+ def test_fpgrowth(self):
+ data = [["a", "b", "c"], ["a", "b", "d", "e"], ["a", "c", "e"],
["a", "c", "f"]]
+ rdd = self.sc.parallelize(data, 2)
+ model = FPGrowth.train(rdd, 0.6, 2)
--- End diff --
This line is useless.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
