use hasWeight to construct the initial graph

vesoft-inc · Nov 4, 2021 · 6956f34 · 6956f34
1 parent 8911233
commit 6956f34
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(