{"action":"created","comment":{"url":"https://api.github.com/repos/apache/spark/pulls/comments/27068205","id":27068205,"diff_hunk":"@@ -83,5 +95,81 @@ class GaussianMixtureModel(\n       p(i) /= pSum\n     }\n     p\n-  }  \n+  }\n+}\n+\n+@Experimental\n+object GaussianMixtureModel extends Loader[GaussianMixtureModel] {\n+\n+  private object SaveLoadV1_0 {\n+\n+    case class Data(weight: Double, mu: Vector, sigma: Matrix)\n+\n+    def formatVersionV1_0 = \"1.0\"\n+\n+    def classNameV1_0 = \"org.apache.spark.mllib.clustering.GaussianMixtureModel\"\n+\n+    def save(\n+        sc: SparkContext,\n+        path: String,\n+        weights: Array[Double],\n+        gaussians: Array[MultivariateGaussian]): Unit = {\n+\n+      val sqlContext = new SQLContext(sc)\n+      import sqlContext.implicits._\n+\n+      // Create JSON metadata.\n+      val metadata = compact(render\n+        ((\"class\" -> classNameV1_0) ~ (\"version\" -> formatVersionV1_0) ~ (\"k\" -> weights.length)))\n+      sc.parallelize(Seq(metadata), 1).saveAsTextFile(Loader.metadataPath(path))\n+\n+      // Create Parquet data.\n+      val dataArray = Array.tabulate(weights.length){ i =>\n+        Data(weights(i), gaussians(i).mu, gaussians(i).sigma)\n+      }\n+      val dataRDD: DataFrame = sc.parallelize(dataArray, 1).toDF()\n+      dataRDD.saveAsParquetFile(Loader.dataPath(path))\n+    }\n+\n+    def load(sc: SparkContext, path: String) : GaussianMixtureModel = {\n+      val datapath = Loader.dataPath(path)\n+      val sqlContext = new SQLContext(sc)\n+      val dataRDD = sqlContext.parquetFile(datapath)\n+      val numGaussians = dataRDD.count().toInt","path":"mllib/src/main/scala/org/apache/spark/mllib/clustering/GaussianMixtureModel.scala","position":84,"original_position":84,"commit_id":"e7a14cbc3f2db51ba6270c4798eb9bfe55a97cfa","original_commit_id":"e7a14cbc3f2db51ba6270c4798eb9bfe55a97cfa","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":"It is not necessary to call `count()`. You can call `val numGaussians = dataArray.length` after `collect()`.\r\n","created_at":"2015-03-24T20:15:46Z","updated_at":"2015-03-24T20:15:46Z","html_url":"https://github.com/apache/spark/pull/4986#discussion_r27068205","pull_request_url":"https://api.github.com/repos/apache/spark/pulls/4986","_links":{"self":{"href":"https://api.github.com/repos/apache/spark/pulls/comments/27068205"},"html":{"href":"https://github.com/apache/spark/pull/4986#discussion_r27068205"},"pull_request":{"href":"https://api.github.com/repos/apache/spark/pulls/4986"}}},"pull_request":{"url":"https://api.github.com/repos/apache/spark/pulls/4986","id":30980403,"html_url":"https://github.com/apache/spark/pull/4986","diff_url":"https://github.com/apache/spark/pull/4986.diff","patch_url":"https://github.com/apache/spark/pull/4986.patch","issue_url":"https://api.github.com/repos/apache/spark/issues/4986","number":4986,"state":"open","locked":false,"title":"[SPARK-5987] [MLlib] Save/load for GaussianMixtureModels","user":{"login":"MechCoder","id":1867024,"avatar_url":"https://avatars.githubusercontent.com/u/1867024?v=3","gravatar_id":"","url":"https://api.github.com/users/MechCoder","html_url":"https://github.com/MechCoder","followers_url":"https://api.github.com/users/MechCoder/followers","following_url":"https://api.github.com/users/MechCoder/following{/other_user}","gists_url":"https://api.github.com/users/MechCoder/gists{/gist_id}","starred_url":"https://api.github.com/users/MechCoder/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/MechCoder/subscriptions","organizations_url":"https://api.github.com/users/MechCoder/orgs","repos_url":"https://api.github.com/users/MechCoder/repos","events_url":"https://api.github.com/users/MechCoder/events{/privacy}","received_events_url":"https://api.github.com/users/MechCoder/received_events","type":"User","site_admin":false},"body":"Should be self explanatory.","created_at":"2015-03-11T20:11:10Z","updated_at":"2015-03-24T20:15:46Z","closed_at":null,"merged_at":null,"merge_commit_sha":"5766b8f786e86df7a1575dc292fe2ba4f10e0755","assignee":null,"milestone":null,"commits_url":"https://api.github.com/repos/apache/spark/pulls/4986/commits","review_comments_url":"https://api.github.com/repos/apache/spark/pulls/4986/comments","review_comment_url":"https://api.github.com/repos/apache/spark/pulls/comments{/number}","comments_url":"https://api.github.com/repos/apache/spark/issues/4986/comments","statuses_url":"https://api.github.com/repos/apache/spark/statuses/e7a14cbc3f2db51ba6270c4798eb9bfe55a97cfa","head":{"label":"MechCoder:spark-5987","ref":"spark-5987","sha":"e7a14cbc3f2db51ba6270c4798eb9bfe55a97cfa","user":{"login":"MechCoder","id":1867024,"avatar_url":"https://avatars.githubusercontent.com/u/1867024?v=3","gravatar_id":"","url":"https://api.github.com/users/MechCoder","html_url":"https://github.com/MechCoder","followers_url":"https://api.github.com/users/MechCoder/followers","following_url":"https://api.github.com/users/MechCoder/following{/other_user}","gists_url":"https://api.github.com/users/MechCoder/gists{/gist_id}","starred_url":"https://api.github.com/users/MechCoder/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/MechCoder/subscriptions","organizations_url":"https://api.github.com/users/MechCoder/orgs","repos_url":"https://api.github.com/users/MechCoder/repos","events_url":"https://api.github.com/users/MechCoder/events{/privacy}","received_events_url":"https://api.github.com/users/MechCoder/received_events","type":"User","site_admin":false},"repo":{"id":23950071,"name":"spark","full_name":"MechCoder/spark","owner":{"login":"MechCoder","id":1867024,"avatar_url":"https://avatars.githubusercontent.com/u/1867024?v=3","gravatar_id":"","url":"https://api.github.com/users/MechCoder","html_url":"https://github.com/MechCoder","followers_url":"https://api.github.com/users/MechCoder/followers","following_url":"https://api.github.com/users/MechCoder/following{/other_user}","gists_url":"https://api.github.com/users/MechCoder/gists{/gist_id}","starred_url":"https://api.github.com/users/MechCoder/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/MechCoder/subscriptions","organizations_url":"https://api.github.com/users/MechCoder/orgs","repos_url":"https://api.github.com/users/MechCoder/repos","events_url":"https://api.github.com/users/MechCoder/events{/privacy}","received_events_url":"https://api.github.com/users/MechCoder/received_events","type":"User","site_admin":false},"private":false,"html_url":"https://github.com/MechCoder/spark","description":"Mirror of Apache Spark","fork":true,"url":"https://api.github.com/repos/MechCoder/spark","forks_url":"https://api.github.com/repos/MechCoder/spark/forks","keys_url":"https://api.github.com/repos/MechCoder/spark/keys{/key_id}","collaborators_url":"https://api.github.com/repos/MechCoder/spark/collaborators{/collaborator}","teams_url":"https://api.github.com/repos/MechCoder/spark/teams","hooks_url":"https://api.github.com/repos/MechCoder/spark/hooks","issue_events_url":"https://api.github.com/repos/MechCoder/spark/issues/events{/number}","events_url":"https://api.github.com/repos/MechCoder/spark/events","assignees_url":"https://api.github.com/repos/MechCoder/spark/assignees{/user}","branches_url":"https://api.github.com/repos/MechCoder/spark/branches{/branch}","tags_url":"https://api.github.com/repos/MechCoder/spark/tags","blobs_url":"https://api.github.com/repos/MechCoder/spark/git/blobs{/sha}","git_tags_url":"https://api.github.com/repos/MechCoder/spark/git/tags{/sha}","git_refs_url":"https://api.github.com/repos/MechCoder/spark/git/refs{/sha}","trees_url":"https://api.github.com/repos/MechCoder/spark/git/trees{/sha}","statuses_url":"https://api.github.com/repos/MechCoder/spark/statuses/{sha}","languages_url":"https://api.github.com/repos/MechCoder/spark/languages","stargazers_url":"https://api.github.com/repos/MechCoder/spark/stargazers","contributors_url":"https://api.github.com/repos/MechCoder/spark/contributors","subscribers_url":"https://api.github.com/repos/MechCoder/spark/subscribers","subscription_url":"https://api.github.com/repos/MechCoder/spark/subscription","commits_url":"https://api.github.com/repos/MechCoder/spark/commits{/sha}","git_commits_url":"https://api.github.com/repos/MechCoder/spark/git/commits{/sha}","comments_url":"https://api.github.com/repos/MechCoder/spark/comments{/number}","issue_comment_url":"https://api.github.com/repos/MechCoder/spark/issues/comments{/number}","contents_url":"https://api.github.com/repos/MechCoder/spark/contents/{+path}","compare_url":"https://api.github.com/repos/MechCoder/spark/compare/{base}...{head}","merges_url":"https://api.github.com/repos/MechCoder/spark/merges","archive_url":"https://api.github.com/repos/MechCoder/spark/{archive_format}{/ref}","downloads_url":"https://api.github.com/repos/MechCoder/spark/downloads","issues_url":"https://api.github.com/repos/MechCoder/spark/issues{/number}","pulls_url":"https://api.github.com/repos/MechCoder/spark/pulls{/number}","milestones_url":"https://api.github.com/repos/MechCoder/spark/milestones{/number}","notifications_url":"https://api.github.com/repos/MechCoder/spark/notifications{?since,all,participating}","labels_url":"https://api.github.com/repos/MechCoder/spark/labels{/name}","releases_url":"https://api.github.com/repos/MechCoder/spark/releases{/id}","created_at":"2014-09-12T06:52:44Z","updated_at":"2015-03-24T12:33:29Z","pushed_at":"2015-03-24T19:30:39Z","git_url":"git://github.com/MechCoder/spark.git","ssh_url":"git@github.com:MechCoder/spark.git","clone_url":"https://github.com/MechCoder/spark.git","svn_url":"https://github.com/MechCoder/spark","homepage":null,"size":94251,"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":"e5d2c37c68ac00a57c2542e62d1c5b4ca267c89e","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-03-24T19:28:20Z","pushed_at":"2015-03-24T19:54:32Z","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":1432017,"stargazers_count":3351,"watchers_count":3351,"language":"Scala","has_issues":false,"has_downloads":true,"has_wiki":false,"has_pages":false,"forks_count":2886,"mirror_url":"git://git.apache.org/spark.git","open_issues_count":299,"forks":2886,"open_issues":299,"watchers":3351,"default_branch":"master"}},"_links":{"self":{"href":"https://api.github.com/repos/apache/spark/pulls/4986"},"html":{"href":"https://github.com/apache/spark/pull/4986"},"issue":{"href":"https://api.github.com/repos/apache/spark/issues/4986"},"comments":{"href":"https://api.github.com/repos/apache/spark/issues/4986/comments"},"review_comments":{"href":"https://api.github.com/repos/apache/spark/pulls/4986/comments"},"review_comment":{"href":"https://api.github.com/repos/apache/spark/pulls/comments{/number}"},"commits":{"href":"https://api.github.com/repos/apache/spark/pulls/4986/commits"},"statuses":{"href":"https://api.github.com/repos/apache/spark/statuses/e7a14cbc3f2db51ba6270c4798eb9bfe55a97cfa"}}}}