Add proc for creating node regression pipelines

Co-authored-by: Mats Rydberg <mats@neotechnology.com>

Author: FlorentinD

doc/antora/content-nav.adoc

@@ -104,6 +104,7 @@
 **** xref:machine-learning/node-property-prediction/nodeclassification-pipelines/config/index.adoc[]
 **** xref:machine-learning/node-property-prediction/nodeclassification-pipelines/training/index.adoc[]
 **** xref:machine-learning/node-property-prediction/nodeclassification-pipelines/predict/index.adoc[]
+*** xref:machine-learning/node-property-prediction/noderegression-pipelines/index.adoc[]
 ** xref:machine-learning/linkprediction-pipelines/index.adoc[]
 *** xref:machine-learning/linkprediction-pipelines/config/index.adoc[]
 *** xref:machine-learning/linkprediction-pipelines/training/index.adoc[]

doc/asciidoc/machine-learning/node-property-prediction/node-property-prediction.adoc

@@ -11,5 +11,9 @@ The Neo4j Graph Data Science library support the following node property predict
 
 * Beta
 ** <<nodeclassification-pipelines>>
+* Alpha
+** <<noderegression-pipelines>>
 
 include::nodeclassification-pipeline/nodeclassification.adoc[leveloffset=+1]
+include::noderegression-pipeline/noderegression.adoc[leveloffset=+1]
+

doc/asciidoc/machine-learning/node-property-prediction/noderegression-pipeline/noderegression.adoc

@@ -0,0 +1,4 @@
+[[noderegression-pipelines]]
+= Node regression pipelines
+
+// TODO add config content

doc/asciidoc/operations-reference/appendix-a-machine-learning.adoc

@@ -62,6 +62,8 @@ include::model-operation-references.adoc[]
 .2+<.^| <<nodeclassification-pipelines, Node Classification Pipeline>>
 | `gds.alpha.pipeline.nodeClassification.addRandomForest`
 | `gds.alpha.pipeline.nodeClassification.configureAutoTuning`
+.1+<.^| <<noderegression-pipelines, Node Regression Pipeline>>
+| `gds.alpha.pipeline.nodeRegression.create`
 |===
 
 

doc/docbook/content-map.xml

@@ -304,6 +304,8 @@
                 <d:tocentry linkend="nodeclassification-pipelines-predict"><?dbhtml filename="machine-learning/node-property-prediction/nodeclassification-pipelines/predict/index.html"?>
                 </d:tocentry>
             </d:tocentry>
+            <d:tocentry linkend="noderegression-pipelines"><?dbhtml filename="machine-learning/node-property-prediction/noderegression-pipelines/index.html"?>
+            </d:tocentry>
         </d:tocentry>
 
         <d:tocentry linkend="linkprediction-pipelines"><?dbhtml filename="machine-learning/linkprediction-pipelines/index.html"?>

open-packaging/src/test/java/org/neo4j/gds/OpenGdsProcedureSmokeTest.java

@@ -94,6 +94,8 @@ class OpenGdsProcedureSmokeTest extends BaseProcTest {
         "gds.beta.pipeline.linkPrediction.train",
         "gds.beta.pipeline.linkPrediction.train.estimate",
 
+        "gds.alpha.pipeline.nodeRegression.create",
+
         "gds.beta.pipeline.nodeClassification.selectFeatures",
         "gds.beta.pipeline.nodeClassification.addNodeProperty",
         "gds.beta.pipeline.nodeClassification.addLogisticRegression",
@@ -444,7 +446,7 @@ void countShouldMatch() {
         );
 
         // If you find yourself updating this count, please also update the count in SmokeTest.kt
-        int expectedCount = 304;
+        int expectedCount = 305;
         assertEquals(
             expectedCount,
             registeredProcedures.size(),

proc/machine-learning/src/main/java/org/neo4j/gds/ml/pipeline/node/regression/configure/NodeRegressionPipelineCreateProc.java

@@ -0,0 +1,48 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.ml.pipeline.node.regression.configure;
+
+import org.neo4j.gds.BaseProc;
+import org.neo4j.gds.core.StringIdentifierValidations;
+import org.neo4j.gds.ml.pipeline.PipelineCatalog;
+import org.neo4j.gds.ml.pipeline.node.NodePipelineInfoResult;
+import org.neo4j.gds.ml.pipeline.nodePipeline.regression.NodeRegressionTrainingPipeline;
+import org.neo4j.procedure.Description;
+import org.neo4j.procedure.Name;
+import org.neo4j.procedure.Procedure;
+
+import java.util.stream.Stream;
+
+import static org.neo4j.procedure.Mode.READ;
+
+public class NodeRegressionPipelineCreateProc extends BaseProc {
+
+    @Procedure(name = "gds.alpha.pipeline.nodeRegression.create", mode = READ)
+    @Description("Creates a node regression training pipeline in the pipeline catalog.")
+    public Stream<NodePipelineInfoResult> create(@Name("pipelineName") String pipelineName) {
+        StringIdentifierValidations.validateNoWhiteCharacter(pipelineName, "pipelineName");
+
+        var pipeline = new NodeRegressionTrainingPipeline();
+
+        PipelineCatalog.set(username(), pipelineName, pipeline);
+
+        return Stream.of(new NodePipelineInfoResult(pipelineName, pipeline));
+    }
+}

proc/machine-learning/src/test/java/org/neo4j/gds/ml/pipeline/node/regression/configure/NodeRegressionPipelineCreateProcTest.java

@@ -0,0 +1,58 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.ml.pipeline.node.regression.configure;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.neo4j.gds.BaseProcTest;
+import org.neo4j.gds.ml.models.TrainingMethod;
+import org.neo4j.gds.ml.pipeline.AutoTuningConfig;
+import org.neo4j.gds.ml.pipeline.PipelineCatalog;
+import org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictionSplitConfig;
+
+import java.util.List;
+import java.util.Map;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class NodeRegressionPipelineCreateProcTest extends BaseProcTest {
+
+    @BeforeEach
+    void setUp() throws Exception {
+        registerProcedures(NodeRegressionPipelineCreateProc.class);
+    }
+
+    @Test
+    void createPipeline() {
+        assertCypherResult("CALL gds.alpha.pipeline.nodeRegression.create('p')", List.of(Map.of(
+            "name", "p",
+            "nodePropertySteps", List.of(),
+            "featureProperties", List.of(),
+            "splitConfig", NodePropertyPredictionSplitConfig.DEFAULT_CONFIG.toMap(),
+            "autoTuningConfig", AutoTuningConfig.DEFAULT_CONFIG.toMap(),
+            "parameterSpace", Map.of(
+                TrainingMethod.LinearRegression.name(), List.of(),
+                TrainingMethod.RandomForest.name(), List.of()
+            )
+        )));
+
+        assertThat(PipelineCatalog.exists(getUsername(), "p")).isTrue();
+    }
+}