Merge pull request #7 from Nicole00/betweenness

use hasWeight to construct the initial graph
vesoft-inc · Nov 8, 2021 · 3cf85fb · 3cf85fb
2 parents 2e95c3a + 6956f34
commit 3cf85fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala b/...-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala
@@ -30,7 +30,7 @@ object BetweennessCentralityAlgo {
             betweennessConfig: BetweennessConfig,
             hasWeight: Boolean): DataFrame = {
 
-    val graph: Graph[None.type, Double] = NebulaUtil.loadInitGraph(dataset, false)
+    val graph: Graph[None.type, Double] = NebulaUtil.loadInitGraph(dataset, hasWeight)
     val betweennessGraph                = execute(graph, betweennessConfig.maxIter, hasWeight)
 
     val schema = StructType(