Merge pull request #5330 from vnickolov/leiden-algo-benchmark

leiden algo benchmark

Author: vnickolov

algo/src/main/java/org/neo4j/gds/leiden/Leiden.java

@@ -50,7 +50,15 @@ public class Leiden extends Algorithm<HugeLongArray> {
     private final int concurrency;
     private final long seed;
 
-    Leiden(Graph graph, int maxIterations, double gamma, double theta, long seed, ProgressTracker progressTracker) {
+    public Leiden(
+        Graph graph,
+        int maxIterations,
+        double gamma,
+        double theta,
+        long seed,
+        int concurrency,
+        ProgressTracker progressTracker
+    ) {
         super(progressTracker);
         this.rootGraph = graph;
         this.maxIterations = maxIterations;
@@ -60,7 +68,8 @@ public class Leiden extends Algorithm<HugeLongArray> {
 
         // TODO: Pass these two as parameters
         this.executorService = Pools.DEFAULT;
-        this.concurrency = 4;
+        this.concurrency = concurrency;
+
         this.dendrograms = new HugeLongArray[maxIterations];
     }
 

algo/src/test/java/org/neo4j/gds/leiden/FootballTest.java

@@ -52,7 +52,7 @@ class FootballTest {
     @ValueSource(longs = {99999, 25, 323, 405, 58, 61, 7, 8123, 94, 19})
     void leiden(long seed) {
         var gamma = 1.0 / graph.relationshipCount();
-        Leiden leiden = new Leiden(graph, 5, gamma, 0.01, seed, ProgressTracker.NULL_TRACKER);
+        Leiden leiden = new Leiden(graph, 5, gamma, 0.01, seed, 1, ProgressTracker.NULL_TRACKER);
         var communities = leiden.compute();
         var communitiesMap = LongStream
             .range(0, graph.nodeCount())

algo/src/test/java/org/neo4j/gds/leiden/KarateTest.java

@@ -49,7 +49,7 @@ class KarateTest {
     @ValueSource(longs = {99999, 25, 323, 405, 58, 61, 7, 8123, 94, 19})
     void leiden(long seed) {
         var gamma = 1.0 / graph.relationshipCount();
-        Leiden leiden = new Leiden(graph, 5, gamma, 0.01, seed, ProgressTracker.NULL_TRACKER);
+        Leiden leiden = new Leiden(graph, 5, gamma, 0.01, seed, 1, ProgressTracker.NULL_TRACKER);
         var communities = leiden.compute();
         var communitiesMap = LongStream
             .range(0, graph.nodeCount())

algo/src/test/java/org/neo4j/gds/leiden/LeidenTest.java

@@ -74,7 +74,8 @@ class LeidenTest {
 
     @Test
     void leiden() {
-        Leiden leiden = new Leiden(graph, 2, 1.0 / graph.relationshipCount(), 0.01, 19L, ProgressTracker.NULL_TRACKER);
+        Leiden leiden = new Leiden(graph, 2, 1.0 / graph.relationshipCount(), 0.01, 19L, 1, ProgressTracker.NULL_TRACKER
+        );
         var communities = leiden.compute();
         var communitiesMap = LongStream
             .range(0, graph.nodeCount())