Update memory estimations

Author: breakanalysis

ml/ml-algo/src/test/java/org/neo4j/gds/ml/models/randomforest/RandomForestClassifierTest.java

@@ -237,19 +237,19 @@ void predictOverheadMemoryEstimation(
 
     @ParameterizedTest
     @CsvSource(value = {
-        "     6, 100_000,  10, 10, 1,   1, 0.1, 1.0,  4413594,  5226418",
+        "     6, 100_000,  10, 10, 1,   1, 0.1, 1.0,  4413602, 5226426",
         // Should increase fairly little with more trees if training set big.
-        "    10, 100_000,  10, 10, 1,  10, 0.1, 1.0,  4414242,  6295802",
+        "    10, 100_000,  10, 10, 1,  10, 0.1, 1.0,  4414250, 6295810",
         // Should be capped by number of training examples, despite high max depth.
-        " 8_000,     500,  10, 10, 1,   1, 0.1, 1.0,    23154,   182954",
+        " 8_000,     500,  10, 10, 1,   1, 0.1, 1.0,    23162, 182962",
         // Should increase very little when having more classes.
-        "    10, 100_000, 100, 10, 1,  10, 0.1, 1.0,  4414962,  6296522",
+        "    10, 100_000, 100, 10, 1,  10, 0.1, 1.0,  4414970, 6296530",
         // Should increase very little when using more features for splits.
-        "    10, 100_000, 100, 10, 1,  10, 0.9, 1.0,  4415034,  6296686",
+        "    10, 100_000, 100, 10, 1,  10, 0.9, 1.0,  4415042, 6296694",
         // Should decrease a lot when sampling fewer training examples per tree.
-        "    10, 100_000, 100, 10, 1,  10, 0.1, 0.2,  1204962,  2446522",
+        "    10, 100_000, 100, 10, 1,  10, 0.1, 0.2,  1204970, 2446530",
         // Should almost be x4 when concurrency * 4.
-        "    10, 100_000, 100, 10, 4,  10, 0.1, 1.0, 16457256, 21037256",
+        "    10, 100_000, 100, 10, 4,  10, 0.1, 1.0, 16457264, 21037264",
     })
     void trainMemoryEstimation(
         int maxDepth,

pipeline/src/test/java/org/neo4j/gds/ml/pipeline/linkPipeline/train/LinkPredictionTrainTest.java

@@ -175,15 +175,15 @@ static Stream<Arguments> paramsForEstimationsWithParamSpace() {
                     .build()
                     .toTunableConfig()
                 ),
-                MemoryRange.of(59696, 900016)
+                MemoryRange.of(59704, 900024)
             ),
             Arguments.of(
                 "Default RF and default LR",
                 List.of(
                     LogisticRegressionTrainConfig.DEFAULT.toTunableConfig(),
                     RandomForestTrainerConfig.DEFAULT.toTunableConfig()
                 ),
-                MemoryRange.of(66696, 2731864)
+                MemoryRange.of(66704, 2731872)
             ),
             Arguments.of(
                 "Default RF and default LR with range",
@@ -194,7 +194,7 @@ static Stream<Arguments> paramsForEstimationsWithParamSpace() {
                     ),
                     RandomForestTrainerConfig.DEFAULT.toTunableConfig()
                 ),
-                MemoryRange.of(66696, 2731864)
+                MemoryRange.of(66704, 2731872)
             ),
             Arguments.of(
                 "Default RF and default LR with batch size range",

proc/machine-learning/src/test/java/org/neo4j/gds/ml/nodemodels/pipeline/NodeClassificationTrainPipelineExecutorTest.java

@@ -273,7 +273,7 @@ public static Stream<Arguments> trainerMethodConfigs() {
             ),
             Arguments.of(
                 List.of(RandomForestTrainerConfig.DEFAULT.toTunableConfig()),
-                MemoryRange.of(106578L, 223350L)
+                MemoryRange.of(106586L, 223358L)
             ),
             Arguments.of(
                 List.of(LogisticRegressionTrainConfig.DEFAULT.toTunableConfig(), RandomForestTrainerConfig.DEFAULT.toTunableConfig()),