Package org.neo4j.gds.ml.models

Class ClassifierFactory

java.lang.Object

org.neo4j.gds.ml.models.ClassifierFactory

public final class ClassifierFactory
extends Object

Method Summary

Modifier and Type	Method	Description
static Classifier	create(Classifier.ClassifierData classifierData)
static org.neo4j.gds.core.utils.mem.MemoryEstimation	dataMemoryEstimation(TrainerConfig trainerConfig, LongUnaryOperator numberOfTrainingSamples, int numberOfClasses, int featureDimension, boolean isReduced)
static org.neo4j.gds.core.utils.mem.MemoryRange	runtimeOverheadMemoryEstimation(org.neo4j.gds.ml.api.TrainingMethod method, int batchSize, int numberOfClasses, int featureDimension, boolean isReduced)

Methods inherited from class java.lang.Object

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

Method Details

create

public static Classifier create(Classifier.ClassifierData classifierData)

runtimeOverheadMemoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryRange runtimeOverheadMemoryEstimation(org.neo4j.gds.ml.api.TrainingMethod method,
 int batchSize,
 int numberOfClasses,
 int featureDimension,
 boolean isReduced)

dataMemoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryEstimation dataMemoryEstimation(TrainerConfig trainerConfig,
 LongUnaryOperator numberOfTrainingSamples,
 int numberOfClasses,
 int featureDimension,
 boolean isReduced)