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.