diff options
| author | zc he <farseer90718@gmail.com> | 2015-08-14 21:28:50 -0700 |
|---|---|---|
| committer | Reynold Xin <rxin@databricks.com> | 2015-08-14 21:28:50 -0700 |
| commit | 71a3af8a94f900a26ac7094f22ec1216cab62e15 (patch) | |
| tree | b243e30e29e43b3d4876b4b4b2a4ce6fe1492e45 /graphx | |
| parent | 609ce3c07d4962a9242e488ad0ed48c183896802 (diff) | |
| download | spark-71a3af8a94f900a26ac7094f22ec1216cab62e15.tar.gz spark-71a3af8a94f900a26ac7094f22ec1216cab62e15.tar.bz2 spark-71a3af8a94f900a26ac7094f22ec1216cab62e15.zip | |
[SPARK-9960] [GRAPHX] sendMessage type fix in LabelPropagation.scala
Author: zc he <farseer90718@gmail.com>
Closes #8188 from farseer90718/farseer-patch-1.
Diffstat (limited to 'graphx')
| -rw-r--r-- | graphx/src/main/scala/org/apache/spark/graphx/lib/LabelPropagation.scala | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/graphx/src/main/scala/org/apache/spark/graphx/lib/LabelPropagation.scala b/graphx/src/main/scala/org/apache/spark/graphx/lib/LabelPropagation.scala index 2bcf8684b8..a3ad6bed1c 100644 --- a/graphx/src/main/scala/org/apache/spark/graphx/lib/LabelPropagation.scala +++ b/graphx/src/main/scala/org/apache/spark/graphx/lib/LabelPropagation.scala @@ -43,7 +43,7 @@ object LabelPropagation { */ def run[VD, ED: ClassTag](graph: Graph[VD, ED], maxSteps: Int): Graph[VertexId, ED] = { val lpaGraph = graph.mapVertices { case (vid, _) => vid } - def sendMessage(e: EdgeTriplet[VertexId, ED]): Iterator[(VertexId, Map[VertexId, VertexId])] = { + def sendMessage(e: EdgeTriplet[VertexId, ED]): Iterator[(VertexId, Map[VertexId, Long])] = { Iterator((e.srcId, Map(e.dstAttr -> 1L)), (e.dstId, Map(e.srcAttr -> 1L))) } def mergeMessage(count1: Map[VertexId, Long], count2: Map[VertexId, Long]) |