Package org.neo4j.gds.ml.nodeClassification

Class NodeClassificationPredict

java.lang.Object

org.neo4j.gds.ml.nodeClassification.NodeClassificationPredict

public class NodeClassificationPredict
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	NodeClassificationPredict.NodeClassificationResult

Constructor Summary

Constructors

Constructor	Description
NodeClassificationPredict(Classifier classifier, Features features, int batchSize, int concurrency, boolean produceProbabilities, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.termination.TerminationFlag terminationFlag)

Method Summary

Modifier and Type	Method	Description
NodeClassificationPredict.NodeClassificationResult	compute()
static org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(boolean produceProbabilities, int batchSize, int featureCount, int classCount)
static org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimationWithDerivedBatchSize(org.neo4j.gds.ml.api.TrainingMethod method, boolean produceProbabilities, int minBatchSize, int featureCount, int classCount, boolean isReduced)
static org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(long nodeCount)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

NodeClassificationPredict

public NodeClassificationPredict(Classifier classifier,
 Features features,
 int batchSize,
 int concurrency,
 boolean produceProbabilities,
 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
 org.neo4j.gds.termination.TerminationFlag terminationFlag)

Method Details

progressTask

public static org.neo4j.gds.core.utils.progress.tasks.Task progressTask(long nodeCount)

memoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(boolean produceProbabilities,
 int batchSize,
 int featureCount,
 int classCount)

memoryEstimationWithDerivedBatchSize

public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimationWithDerivedBatchSize(org.neo4j.gds.ml.api.TrainingMethod method,
 boolean produceProbabilities,
 int minBatchSize,
 int featureCount,
 int classCount,
 boolean isReduced)

compute

public NodeClassificationPredict.NodeClassificationResult compute()