diff --git a/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala b/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala index e6284e1..b8ba056 100644 --- a/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/lib/BetweennessCentralityAlgo.scala +++ b/nebula-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(