ODIn/spark-instrumented-optimizer
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

added pregel pagerank

Browse source
This commit is contained in:
Joseph E. Gonzalez 2013-04-04 19:31:14 -07:00
parent d510045d8c
commit ad908f7545
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
graph/src/main/scala/spark/graph/Analytics.scala
View file

@ -40,6 +40,22 @@ object Analytics {
}
/**
* Compute the PageRank of a graph returning the pagerank of each vertex as an RDD
*/
def pregelPagerank[VD: Manifest, ED: Manifest](graph: Graph[VD, ED], numIter: Int) = {
// Compute the out degree of each vertex
val pagerankGraph = graph.updateVertices[Int, (Int, Float)](graph.outDegrees,
(vertex, degIter) => (degIter.sum, 1.0F)
)
Pregel.iterate[(Int, Float), ED, Float](pagerankGraph)(
(vertex, a: Float) => (vertex.data._1, (0.15F + 0.85F * a)), // apply
(me_id, edge) => Some(edge.src.data._2 / edge.src.data._1), // gather
(a: Float, b: Float) => a + b, // merge
numIter).mapVertices{ case Vertex(id, (outDeg, r)) => Vertex(id, r) }
}
/**
* Compute the PageRank of a graph returning the pagerank of each vertex as an RDD
*/