4747
4848import static org .neo4j .gds .executor .ExecutionMode .MUTATE_NODE_PROPERTY ;
4949import static org .neo4j .gds .ml .PipelineCompanion .prepareTrainConfig ;
50+ import static org .neo4j .gds .ml .pipeline .node .classification .NodeClassificationPipelineCompanion .ESTIMATE_PREDICT_DESCRIPTION ;
51+ import static org .neo4j .gds .ml .pipeline .node .classification .NodeClassificationPipelineCompanion .PREDICT_DESCRIPTION ;
5052
51- @ GdsCallable (name = "gds.beta.pipeline.nodeClassification.predict.mutate" , description = org . neo4j . gds . ml . pipeline . node . classification . NodeClassificationPipelineCompanion . PREDICT_DESCRIPTION , executionMode = MUTATE_NODE_PROPERTY )
53+ @ GdsCallable (name = "gds.beta.pipeline.nodeClassification.predict.mutate" , description = PREDICT_DESCRIPTION , executionMode = MUTATE_NODE_PROPERTY )
5254public class NodeClassificationPipelineMutateProc
5355 extends MutatePropertyProc <
5456 NodeClassificationPredictPipelineExecutor ,
@@ -57,7 +59,7 @@ public class NodeClassificationPipelineMutateProc
5759 NodeClassificationPredictPipelineMutateConfig >
5860{
5961 @ Procedure (name = "gds.beta.pipeline.nodeClassification.predict.mutate" , mode = Mode .READ )
60- @ Description (org . neo4j . gds . ml . pipeline . node . classification . NodeClassificationPipelineCompanion . PREDICT_DESCRIPTION )
62+ @ Description (PREDICT_DESCRIPTION )
6163 public Stream <MutateResult > mutate (
6264 @ Name (value = "graphName" ) String graphName ,
6365 @ Name (value = "configuration" , defaultValue = "{}" ) Map <String , Object > configuration
@@ -67,7 +69,7 @@ public Stream<MutateResult> mutate(
6769 }
6870
6971 @ Procedure (name = "gds.beta.pipeline.nodeClassification.predict.mutate.estimate" , mode = Mode .READ )
70- @ Description (org . neo4j . gds . ml . pipeline . node . classification . NodeClassificationPipelineCompanion . ESTIMATE_PREDICT_DESCRIPTION )
72+ @ Description (ESTIMATE_PREDICT_DESCRIPTION )
7173 public Stream <MemoryEstimateResult > estimate (
7274 @ Name (value = "graphNameOrConfiguration" ) Object graphNameOrConfiguration ,
7375 @ Name (value = "algoConfiguration" ) Map <String , Object > algoConfiguration
@@ -98,7 +100,7 @@ protected List<NodeProperty> nodePropertyList(ComputationResult<NodeClassificati
98100 var nodeProperties = new ArrayList <NodeProperty >();
99101 nodeProperties .add (NodeProperty .of (mutateProperty , classProperties ));
100102
101- result .predictedProbabilities ().ifPresent (( probabilityProperties ) -> {
103+ result .predictedProbabilities ().ifPresent (probabilityProperties -> {
102104 var properties = new DoubleArrayNodePropertyValues () {
103105 @ Override
104106 public long size () {
0 commit comments