Class NodeClassificationPredictConsumer

java.lang.Object
org.neo4j.gds.ml.nodeClassification.NodeClassificationPredictConsumer
All Implemented Interfaces:
Consumer<org.neo4j.gds.ml.core.batch.Batch>

public class NodeClassificationPredictConsumer extends Object implements Consumer<org.neo4j.gds.ml.core.batch.Batch>
Consumes a BatchQueue containing long indices into a nodeIds LongArrayAccessor. The consumer will apply node classification to the ids in nodeIds and write them in the same order into predictedClasses. If predictedProbabilities is non-null, the predicted probabilities will also be written into it.
  • Method Details

    • accept

      public void accept(org.neo4j.gds.ml.core.batch.Batch batch)
      Specified by:
      accept in interface Consumer<org.neo4j.gds.ml.core.batch.Batch>