Make various LP tasks use setSteps

Co-Authored-By: Adam Schill Collberg<adam.schill.collberg@protonmail.com>

Author: breakanalysis

ml/ml-algo/src/main/java/org/neo4j/gds/ml/metrics/SignedProbabilities.java

@@ -119,7 +119,7 @@ public static SignedProbabilities computeFromLabeledData(
         TerminationFlag terminationFlag,
         ProgressTracker progressTracker
     ) {
-        progressTracker.setVolume(features.size());
+        progressTracker.setSteps(features.size());
 
         var signedProbabilities = SignedProbabilities.create(evaluationQueue.totalSize());
 
@@ -134,7 +134,7 @@ public static SignedProbabilities computeFromLabeledData(
 
                     signedProbabilities.add(probabilityOfPositiveEdge, isEdge);
                 }
-                progressTracker.logProgress(batch.size());
+                progressTracker.logSteps(batch.size());
             },
             terminationFlag
         );

pipeline/src/main/java/org/neo4j/gds/ml/pipeline/linkPipeline/BatchLinkFeatureExtractor.java

@@ -61,7 +61,7 @@ public void run() {
             }));
         });
 
-        progressTracker.logProgress(partition.totalDegree());
+        progressTracker.logSteps(partition.totalDegree());
     }
 }
 

pipeline/src/main/java/org/neo4j/gds/ml/pipeline/linkPipeline/train/LinkFeaturesAndLabelsExtractor.java

@@ -79,7 +79,7 @@ static FeaturesAndLabels extractFeaturesAndLabels(
         ProgressTracker progressTracker,
         TerminationFlag terminationFlag
     ) {
-        progressTracker.setVolume(graph.relationshipCount() * 2);
+        progressTracker.setSteps(graph.relationshipCount() * 2);
         var features = LinkFeatureExtractor.extractFeatures(
             graph,
             featureSteps,
@@ -125,7 +125,7 @@ private static HugeLongArray extractLabels(
                     }
                     return true;
                 }));
-                progressTracker.logProgress(partition.totalDegree());
+                progressTracker.logSteps(partition.totalDegree());
             }
             );
             relationshipOffset.add(partition.totalDegree());

pipeline/src/main/java/org/neo4j/gds/ml/pipeline/linkPipeline/train/LinkPredictionTrain.java

@@ -210,11 +210,10 @@ private void modelSelect(
             config.randomSeed()
         );
 
-        var trainRelationshipCount = trainGraph.relationshipCount();
         int trial = 0;
         while (hyperParameterOptimizer.hasNext()) {
             progressTracker.beginSubTask();
-            progressTracker.setVolume(pipeline.splitConfig().validationFolds() * trainRelationshipCount);
+            progressTracker.setSteps(pipeline.splitConfig().validationFolds());
             var modelParams = hyperParameterOptimizer.next();
             progressTracker.logMessage(formatWithLocale("Method: %s, Parameters: %s", modelParams.method(), modelParams.toMap()));
             var trainStatsBuilder = new ModelStatsBuilder(pipeline.splitConfig().validationFolds());
@@ -249,7 +248,7 @@ private void modelSelect(
                     ProgressTracker.NULL_TRACKER
                 );
 
-                progressTracker.logProgress(trainRelationshipCount);
+                progressTracker.logSteps(1);
             }
 
             // insert the candidates' metrics into trainStats and validationStats