{"action":"created","comment":{"url":"https://api.github.com/repos/apache/spark/pulls/comments/33998097","id":33998097,"diff_hunk":"@@ -0,0 +1,119 @@\n+/*\n+ * Licensed to the Apache Software Foundation (ASF) under one or more\n+ * contributor license agreements.  See the NOTICE file distributed with\n+ * this work for additional information regarding copyright ownership.\n+ * The ASF licenses this file to You under the Apache License, Version 2.0\n+ * (the \"License\"); you may not use this file except in compliance with\n+ * the License.  You may obtain a copy of the License at\n+ *\n+ *    http://www.apache.org/licenses/LICENSE-2.0\n+ *\n+ * Unless required by applicable law or agreed to in writing, software\n+ * distributed under the License is distributed on an \"AS IS\" BASIS,\n+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n+ * See the License for the specific language governing permissions and\n+ * limitations under the License.\n+ */\n+package org.apache.spark.mllib.fpm\n+\n+import java.lang.{Iterable => JavaIterable}\n+\n+import org.apache.spark.api.java.JavaSparkContext.fakeClassTag\n+import org.apache.spark.api.java.JavaRDD\n+import org.apache.spark.mllib.fpm.FPGrowth.FreqItemset\n+\n+import scala.collection.JavaConverters._\n+import scala.reflect.ClassTag\n+\n+import org.apache.spark.Logging\n+import org.apache.spark.annotation.Experimental\n+import org.apache.spark.mllib.fpm.AssociationRules.Rule\n+import org.apache.spark.rdd.RDD\n+\n+/**\n+ * :: Experimental ::\n+ *\n+ * Generates association rules from a [[RDD[FreqItemset[Item]]]. This method only generates\n+ * association rules which have a single item as the consequent.\n+ */\n+@Experimental\n+class AssociationRules private (\n+    private var minConfidence: Double) extends Logging with Serializable {\n+\n+  /**\n+   * Constructs a default instance with default parameters {minConfidence = 0.5}.\n+   */\n+  def this() = this(0.5)\n+\n+  /**\n+   * Sets the minimal confidence (default: `0.5`).\n+   */\n+  def setMinConfidence(minConfidence: Double): this.type = {\n+    this.minConfidence = minConfidence\n+    this\n+  }\n+\n+  /**\n+   * Computes the association rules with confidence above [[minConfidence]].\n+   * @param model frequent itemset model obtained from [[FPGrowth]]\n+   * @return a [[Set[Rule[Item]]] containing the assocation rules.\n+   */\n+  def run[Item: ClassTag](model: FPGrowthModel[Item]): RDD[Rule[Item]] = {\n+    val freqItemsets = model.freqItemsets\n+    val numTransactions = model.numTransactions\n+\n+    freqItemsets.flatMap { itemset =>\n+      val items = itemset.items\n+      items.map { item =>\n+        // Key using List[Item] because cannot use Array[Item] for map-side combining\n+        items.partition(_ == item) match {\n+          case (consequent, antecedent) => (antecedent.toList, (consequent.toList, itemset.freq))\n+        }\n+      } :+ (items.toList, (Nil, itemset.freq))\n+    }.aggregateByKey(Map[List[Item], Long]().empty)(\n+      { case (acc, (consequent, freq)) => acc + (consequent -> freq) },\n+      { (acc1, acc2) => acc1 ++ acc2 }","path":"mllib/src/main/scala/org/apache/spark/mllib/fpm/AssociationRules.scala","position":75,"original_position":75,"commit_id":"0cc1a6a61cde78bb24eb2890b99f87c4a491f948","original_commit_id":"0cc1a6a61cde78bb24eb2890b99f87c4a491f948","user":{"login":"mengxr","id":829644,"avatar_url":"https://avatars.githubusercontent.com/u/829644?v=3","gravatar_id":"","url":"https://api.github.com/users/mengxr","html_url":"https://github.com/mengxr","followers_url":"https://api.github.com/users/mengxr/followers","following_url":"https://api.github.com/users/mengxr/following{/other_user}","gists_url":"https://api.github.com/users/mengxr/gists{/gist_id}","starred_url":"https://api.github.com/users/mengxr/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/mengxr/subscriptions","organizations_url":"https://api.github.com/users/mengxr/orgs","repos_url":"https://api.github.com/users/mengxr/repos","events_url":"https://api.github.com/users/mengxr/events{/privacy}","received_events_url":"https://api.github.com/users/mengxr/received_events","type":"User","site_admin":false},"body":"`++`","created_at":"2015-07-07T00:26:29Z","updated_at":"2015-07-07T00:26:29Z","html_url":"https://github.com/apache/spark/pull/7005#discussion_r33998097","pull_request_url":"https://api.github.com/repos/apache/spark/pulls/7005","_links":{"self":{"href":"https://api.github.com/repos/apache/spark/pulls/comments/33998097"},"html":{"href":"https://github.com/apache/spark/pull/7005#discussion_r33998097"},"pull_request":{"href":"https://api.github.com/repos/apache/spark/pulls/7005"}}},"pull_request":{"url":"https://api.github.com/repos/apache/spark/pulls/7005","id":38500707,"html_url":"https://github.com/apache/spark/pull/7005","diff_url":"https://github.com/apache/spark/pull/7005.diff","patch_url":"https://github.com/apache/spark/pull/7005.patch","issue_url":"https://api.github.com/repos/apache/spark/issues/7005","number":7005,"state":"open","locked":false,"title":"[SPARK-8559][MLlib] Support Association Rule Generation","user":{"login":"feynmanliang","id":990069,"avatar_url":"https://avatars.githubusercontent.com/u/990069?v=3","gravatar_id":"","url":"https://api.github.com/users/feynmanliang","html_url":"https://github.com/feynmanliang","followers_url":"https://api.github.com/users/feynmanliang/followers","following_url":"https://api.github.com/users/feynmanliang/following{/other_user}","gists_url":"https://api.github.com/users/feynmanliang/gists{/gist_id}","starred_url":"https://api.github.com/users/feynmanliang/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/feynmanliang/subscriptions","organizations_url":"https://api.github.com/users/feynmanliang/orgs","repos_url":"https://api.github.com/users/feynmanliang/repos","events_url":"https://api.github.com/users/feynmanliang/events{/privacy}","received_events_url":"https://api.github.com/users/feynmanliang/received_events","type":"User","site_admin":false},"body":"Distributed generation of single-consequent association rules from a RDD of frequent itemsets. Tests referenced against `R`'s implementation of A Priori in [arules](http://cran.r-project.org/web/packages/arules/index.html).","created_at":"2015-06-25T00:05:24Z","updated_at":"2015-07-07T00:26:29Z","closed_at":null,"merged_at":null,"merge_commit_sha":"d9692e0bc4b24bb46a36856782364f1341a5bfd1","assignee":null,"milestone":null,"commits_url":"https://api.github.com/repos/apache/spark/pulls/7005/commits","review_comments_url":"https://api.github.com/repos/apache/spark/pulls/7005/comments","review_comment_url":"https://api.github.com/repos/apache/spark/pulls/comments{/number}","comments_url":"https://api.github.com/repos/apache/spark/issues/7005/comments","statuses_url":"https://api.github.com/repos/apache/spark/statuses/0cc1a6a61cde78bb24eb2890b99f87c4a491f948","head":{"label":"feynmanliang:fp-association-rules-distributed","ref":"fp-association-rules-distributed","sha":"0cc1a6a61cde78bb24eb2890b99f87c4a491f948","user":{"login":"feynmanliang","id":990069,"avatar_url":"https://avatars.githubusercontent.com/u/990069?v=3","gravatar_id":"","url":"https://api.github.com/users/feynmanliang","html_url":"https://github.com/feynmanliang","followers_url":"https://api.github.com/users/feynmanliang/followers","following_url":"https://api.github.com/users/feynmanliang/following{/other_user}","gists_url":"https://api.github.com/users/feynmanliang/gists{/gist_id}","starred_url":"https://api.github.com/users/feynmanliang/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/feynmanliang/subscriptions","organizations_url":"https://api.github.com/users/feynmanliang/orgs","repos_url":"https://api.github.com/users/feynmanliang/repos","events_url":"https://api.github.com/users/feynmanliang/events{/privacy}","received_events_url":"https://api.github.com/users/feynmanliang/received_events","type":"User","site_admin":false},"repo":{"id":31143142,"name":"spark","full_name":"feynmanliang/spark","owner":{"login":"feynmanliang","id":990069,"avatar_url":"https://avatars.githubusercontent.com/u/990069?v=3","gravatar_id":"","url":"https://api.github.com/users/feynmanliang","html_url":"https://github.com/feynmanliang","followers_url":"https://api.github.com/users/feynmanliang/followers","following_url":"https://api.github.com/users/feynmanliang/following{/other_user}","gists_url":"https://api.github.com/users/feynmanliang/gists{/gist_id}","starred_url":"https://api.github.com/users/feynmanliang/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/feynmanliang/subscriptions","organizations_url":"https://api.github.com/users/feynmanliang/orgs","repos_url":"https://api.github.com/users/feynmanliang/repos","events_url":"https://api.github.com/users/feynmanliang/events{/privacy}","received_events_url":"https://api.github.com/users/feynmanliang/received_events","type":"User","site_admin":false},"private":false,"html_url":"https://github.com/feynmanliang/spark","description":"Mirror of Apache Spark","fork":true,"url":"https://api.github.com/repos/feynmanliang/spark","forks_url":"https://api.github.com/repos/feynmanliang/spark/forks","keys_url":"https://api.github.com/repos/feynmanliang/spark/keys{/key_id}","collaborators_url":"https://api.github.com/repos/feynmanliang/spark/collaborators{/collaborator}","teams_url":"https://api.github.com/repos/feynmanliang/spark/teams","hooks_url":"https://api.github.com/repos/feynmanliang/spark/hooks","issue_events_url":"https://api.github.com/repos/feynmanliang/spark/issues/events{/number}","events_url":"https://api.github.com/repos/feynmanliang/spark/events","assignees_url":"https://api.github.com/repos/feynmanliang/spark/assignees{/user}","branches_url":"https://api.github.com/repos/feynmanliang/spark/branches{/branch}","tags_url":"https://api.github.com/repos/feynmanliang/spark/tags","blobs_url":"https://api.github.com/repos/feynmanliang/spark/git/blobs{/sha}","git_tags_url":"https://api.github.com/repos/feynmanliang/spark/git/tags{/sha}","git_refs_url":"https://api.github.com/repos/feynmanliang/spark/git/refs{/sha}","trees_url":"https://api.github.com/repos/feynmanliang/spark/git/trees{/sha}","statuses_url":"https://api.github.com/repos/feynmanliang/spark/statuses/{sha}","languages_url":"https://api.github.com/repos/feynmanliang/spark/languages","stargazers_url":"https://api.github.com/repos/feynmanliang/spark/stargazers","contributors_url":"https://api.github.com/repos/feynmanliang/spark/contributors","subscribers_url":"https://api.github.com/repos/feynmanliang/spark/subscribers","subscription_url":"https://api.github.com/repos/feynmanliang/spark/subscription","commits_url":"https://api.github.com/repos/feynmanliang/spark/commits{/sha}","git_commits_url":"https://api.github.com/repos/feynmanliang/spark/git/commits{/sha}","comments_url":"https://api.github.com/repos/feynmanliang/spark/comments{/number}","issue_comment_url":"https://api.github.com/repos/feynmanliang/spark/issues/comments{/number}","contents_url":"https://api.github.com/repos/feynmanliang/spark/contents/{+path}","compare_url":"https://api.github.com/repos/feynmanliang/spark/compare/{base}...{head}","merges_url":"https://api.github.com/repos/feynmanliang/spark/merges","archive_url":"https://api.github.com/repos/feynmanliang/spark/{archive_format}{/ref}","downloads_url":"https://api.github.com/repos/feynmanliang/spark/downloads","issues_url":"https://api.github.com/repos/feynmanliang/spark/issues{/number}","pulls_url":"https://api.github.com/repos/feynmanliang/spark/pulls{/number}","milestones_url":"https://api.github.com/repos/feynmanliang/spark/milestones{/number}","notifications_url":"https://api.github.com/repos/feynmanliang/spark/notifications{?since,all,participating}","labels_url":"https://api.github.com/repos/feynmanliang/spark/labels{/name}","releases_url":"https://api.github.com/repos/feynmanliang/spark/releases{/id}","created_at":"2015-02-21T22:13:57Z","updated_at":"2015-02-21T22:14:17Z","pushed_at":"2015-07-06T23:49:39Z","git_url":"git://github.com/feynmanliang/spark.git","ssh_url":"git@github.com:feynmanliang/spark.git","clone_url":"https://github.com/feynmanliang/spark.git","svn_url":"https://github.com/feynmanliang/spark","homepage":null,"size":109207,"stargazers_count":0,"watchers_count":0,"language":"Scala","has_issues":false,"has_downloads":true,"has_wiki":false,"has_pages":false,"forks_count":0,"mirror_url":null,"open_issues_count":0,"forks":0,"open_issues":0,"watchers":0,"default_branch":"master"}},"base":{"label":"apache:master","ref":"master","sha":"b71d3254e50838ccae43bdb0ff186fda25f03152","user":{"login":"apache","id":47359,"avatar_url":"https://avatars.githubusercontent.com/u/47359?v=3","gravatar_id":"","url":"https://api.github.com/users/apache","html_url":"https://github.com/apache","followers_url":"https://api.github.com/users/apache/followers","following_url":"https://api.github.com/users/apache/following{/other_user}","gists_url":"https://api.github.com/users/apache/gists{/gist_id}","starred_url":"https://api.github.com/users/apache/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/apache/subscriptions","organizations_url":"https://api.github.com/users/apache/orgs","repos_url":"https://api.github.com/users/apache/repos","events_url":"https://api.github.com/users/apache/events{/privacy}","received_events_url":"https://api.github.com/users/apache/received_events","type":"Organization","site_admin":false},"repo":{"id":17165658,"name":"spark","full_name":"apache/spark","owner":{"login":"apache","id":47359,"avatar_url":"https://avatars.githubusercontent.com/u/47359?v=3","gravatar_id":"","url":"https://api.github.com/users/apache","html_url":"https://github.com/apache","followers_url":"https://api.github.com/users/apache/followers","following_url":"https://api.github.com/users/apache/following{/other_user}","gists_url":"https://api.github.com/users/apache/gists{/gist_id}","starred_url":"https://api.github.com/users/apache/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/apache/subscriptions","organizations_url":"https://api.github.com/users/apache/orgs","repos_url":"https://api.github.com/users/apache/repos","events_url":"https://api.github.com/users/apache/events{/privacy}","received_events_url":"https://api.github.com/users/apache/received_events","type":"Organization","site_admin":false},"private":false,"html_url":"https://github.com/apache/spark","description":"Mirror of Apache Spark","fork":false,"url":"https://api.github.com/repos/apache/spark","forks_url":"https://api.github.com/repos/apache/spark/forks","keys_url":"https://api.github.com/repos/apache/spark/keys{/key_id}","collaborators_url":"https://api.github.com/repos/apache/spark/collaborators{/collaborator}","teams_url":"https://api.github.com/repos/apache/spark/teams","hooks_url":"https://api.github.com/repos/apache/spark/hooks","issue_events_url":"https://api.github.com/repos/apache/spark/issues/events{/number}","events_url":"https://api.github.com/repos/apache/spark/events","assignees_url":"https://api.github.com/repos/apache/spark/assignees{/user}","branches_url":"https://api.github.com/repos/apache/spark/branches{/branch}","tags_url":"https://api.github.com/repos/apache/spark/tags","blobs_url":"https://api.github.com/repos/apache/spark/git/blobs{/sha}","git_tags_url":"https://api.github.com/repos/apache/spark/git/tags{/sha}","git_refs_url":"https://api.github.com/repos/apache/spark/git/refs{/sha}","trees_url":"https://api.github.com/repos/apache/spark/git/trees{/sha}","statuses_url":"https://api.github.com/repos/apache/spark/statuses/{sha}","languages_url":"https://api.github.com/repos/apache/spark/languages","stargazers_url":"https://api.github.com/repos/apache/spark/stargazers","contributors_url":"https://api.github.com/repos/apache/spark/contributors","subscribers_url":"https://api.github.com/repos/apache/spark/subscribers","subscription_url":"https://api.github.com/repos/apache/spark/subscription","commits_url":"https://api.github.com/repos/apache/spark/commits{/sha}","git_commits_url":"https://api.github.com/repos/apache/spark/git/commits{/sha}","comments_url":"https://api.github.com/repos/apache/spark/comments{/number}","issue_comment_url":"https://api.github.com/repos/apache/spark/issues/comments{/number}","contents_url":"https://api.github.com/repos/apache/spark/contents/{+path}","compare_url":"https://api.github.com/repos/apache/spark/compare/{base}...{head}","merges_url":"https://api.github.com/repos/apache/spark/merges","archive_url":"https://api.github.com/repos/apache/spark/{archive_format}{/ref}","downloads_url":"https://api.github.com/repos/apache/spark/downloads","issues_url":"https://api.github.com/repos/apache/spark/issues{/number}","pulls_url":"https://api.github.com/repos/apache/spark/pulls{/number}","milestones_url":"https://api.github.com/repos/apache/spark/milestones{/number}","notifications_url":"https://api.github.com/repos/apache/spark/notifications{?since,all,participating}","labels_url":"https://api.github.com/repos/apache/spark/labels{/name}","releases_url":"https://api.github.com/repos/apache/spark/releases{/id}","created_at":"2014-02-25T08:00:08Z","updated_at":"2015-07-06T19:54:10Z","pushed_at":"2015-07-07T00:17:27Z","git_url":"git://github.com/apache/spark.git","ssh_url":"git@github.com:apache/spark.git","clone_url":"https://github.com/apache/spark.git","svn_url":"https://github.com/apache/spark","homepage":null,"size":1499773,"stargazers_count":4646,"watchers_count":4646,"language":"Scala","has_issues":false,"has_downloads":true,"has_wiki":false,"has_pages":false,"forks_count":4124,"mirror_url":"git://git.apache.org/spark.git","open_issues_count":341,"forks":4124,"open_issues":341,"watchers":4646,"default_branch":"master"}},"_links":{"self":{"href":"https://api.github.com/repos/apache/spark/pulls/7005"},"html":{"href":"https://github.com/apache/spark/pull/7005"},"issue":{"href":"https://api.github.com/repos/apache/spark/issues/7005"},"comments":{"href":"https://api.github.com/repos/apache/spark/issues/7005/comments"},"review_comments":{"href":"https://api.github.com/repos/apache/spark/pulls/7005/comments"},"review_comment":{"href":"https://api.github.com/repos/apache/spark/pulls/comments{/number}"},"commits":{"href":"https://api.github.com/repos/apache/spark/pulls/7005/commits"},"statuses":{"href":"https://api.github.com/repos/apache/spark/statuses/0cc1a6a61cde78bb24eb2890b99f87c4a491f948"}}}}