Skip to content

[SPARK-24191][ML]Scala Example code for Power Iteration Clustering #21248

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
shahidki31 wants to merge 5 commits into from
Closed

[SPARK-24191][ML]Scala Example code for Power Iteration Clustering #21248

Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
f53a075
Example code for Power Iteration Clustering
shahidki31 May 5, 2018
a718d4a
Example code for Power Iteration Clustering
shahidki31 May 21, 2018
8c3025e
Merge https://github.com/apache/spark into sparkCommit
shahidki31 Jun 7, 2018
d005493
Example code for Power Iteration Clustering
shahidki31 Jun 7, 2018
6b75b8c
Example code for Power Iteration Clustering
shahidki31 Jun 7, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Example code for Power Iteration Clustering
  • Loading branch information
@shahidki31
shahidki31 committed Jun 7, 2018
commit 6b75b8c20d862dba3f7679833a081296d2a2f8a3
14 changes: 2 additions & 12 deletions examples/src/main/scala/org/apache/spark/examples/ml/PowerIterationClusteringExample.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,19 +20,10 @@ package org.apache.spark.examples.ml

// $example on$
import org.apache.spark.ml.clustering.PowerIterationClustering
import org.apache.spark.sql.SparkSession
// $example off$

/**
* An example demonstrating power iteration clustering.
* Run with
* {{{
* bin/run-example ml.PowerIterationClusteringExample
* }}}
*/
import org.apache.spark.sql.SparkSession

object PowerIterationClusteringExample {

def main(args: Array[String]): Unit = {
val spark = SparkSession
.builder
Expand All @@ -48,7 +39,6 @@ object PowerIterationClusteringExample {
(4L, 0L, 0.1)
)).toDF("src", "dst", "weight")

// Trains a PIC model.
val model = new PowerIterationClustering().
setK(2).
setMaxIter(20).
Expand All @@ -58,7 +48,7 @@ object PowerIterationClusteringExample {
val prediction = model.assignClusters(dataset).select("id", "cluster")

// Shows the cluster assignment
prediction.show()
prediction.show(false)
// $example off$

spark.stop()
Expand Down