File tree Expand file tree Collapse file tree 1 file changed +32
-0
lines changed
algorithms-compute-business-facade/src/main/java/org/neo4j/gds/community Expand file tree Collapse file tree 1 file changed +32
-0
lines changed Original file line number Diff line number Diff line change 5555import org .neo4j .gds .louvain .LouvainResult ;
5656import org .neo4j .gds .modularity .ModularityParameters ;
5757import org .neo4j .gds .modularity .ModularityResult ;
58+ import org .neo4j .gds .modularityoptimization .ModularityOptimizationParameters ;
59+ import org .neo4j .gds .modularityoptimization .ModularityOptimizationResult ;
5860import org .neo4j .gds .result .TimedAlgorithmResult ;
5961import org .neo4j .gds .results .ResultTransformerBuilder ;
6062import org .neo4j .gds .triangle .LocalClusteringCoefficientParameters ;
@@ -444,5 +446,35 @@ public <TR> CompletableFuture<TR> modularity(
444446 ).thenApply (resultTransformerBuilder .build (graphResources ));
445447 }
446448
449+ public <TR > CompletableFuture <TR > modularityOptimization (
450+ GraphName graphName ,
451+ GraphParameters graphParameters ,
452+ Optional <String > relationshipProperty ,
453+ ModularityOptimizationParameters parameters ,
454+ JobId jobId ,
455+ boolean logProgress ,
456+ ResultTransformerBuilder <TimedAlgorithmResult <ModularityOptimizationResult >, TR > resultTransformerBuilder
457+ ) {
458+ // Fetch the Graph the algorithm will operate on
459+ var graphResources = graphStoreCatalogService .fetchGraphResources (
460+ graphName ,
461+ graphParameters ,
462+ relationshipProperty ,
463+ SeedPropertyGraphStoreValidation .create (parameters .seedProperty ().orElse (null )),
464+ Optional .empty (),
465+ user ,
466+ databaseId
467+ );
468+ var graph = graphResources .graph ();
469+
470+ return computeFacade .modularityOptimization (
471+ graph ,
472+ parameters ,
473+ jobId ,
474+ logProgress
475+
476+ ).thenApply (resultTransformerBuilder .build (graphResources ));
477+ }
478+
447479
448480}
You can’t perform that action at this time.
0 commit comments