Package org.neo4j.gds.ml.nodeClassification

Class ClassificationMetricComputer

java.lang.Object

org.neo4j.gds.ml.nodeClassification.ClassificationMetricComputer

public final class ClassificationMetricComputer
extends Object

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimateEvaluation(TrainerConfig config, int batchSize, LongUnaryOperator trainSetSize, LongUnaryOperator testSetSize, int fudgedClassCount, int fudgedFeatureCount, boolean isReduced)
static ClassificationMetricComputer	forEvaluationSet(Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray evaluationSet, Classifier classifier, int concurrency, org.neo4j.gds.termination.TerminationFlag terminationFlag, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
double	score(ClassificationMetric metric)

Methods inherited from class java.lang.Object

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

Method Details

score

public double score(ClassificationMetric metric)

forEvaluationSet

public static ClassificationMetricComputer forEvaluationSet(Features features,
 org.neo4j.gds.collections.ha.HugeIntArray labels,
 org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray evaluationSet,
 Classifier classifier,
 int concurrency,
 org.neo4j.gds.termination.TerminationFlag terminationFlag,
 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

estimateEvaluation

public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimateEvaluation(TrainerConfig config,
 int batchSize,
 LongUnaryOperator trainSetSize,
 LongUnaryOperator testSetSize,
 int fudgedClassCount,
 int fudgedFeatureCount,
 boolean isReduced)