Package org.neo4j.gds.ml.models

Class ClassifierTrainerFactory

java.lang.Object

org.neo4j.gds.ml.models.ClassifierTrainerFactory

public final class ClassifierTrainerFactory
extends Object

Method Summary

Modifier and Type	Method	Description
static ClassifierTrainer	create(TrainerConfig config, int numberOfClasses, org.neo4j.gds.termination.TerminationFlag terminationFlag, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel, int concurrency, Optional<Long> randomSeed, boolean reduceClassCount, ModelSpecificMetricsHandler metricsHandler)
static org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(TrainerConfig config, LongUnaryOperator numberOfTrainingExamples, int numberOfClasses, org.neo4j.gds.core.utils.mem.MemoryRange 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 ClassifierTrainer create(TrainerConfig config,
 int numberOfClasses,
 org.neo4j.gds.termination.TerminationFlag terminationFlag,
 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
 org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel,
 int concurrency,
 Optional<Long> randomSeed,
 boolean reduceClassCount,
 ModelSpecificMetricsHandler metricsHandler)

memoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(TrainerConfig config,
 LongUnaryOperator numberOfTrainingExamples,
 int numberOfClasses,
 org.neo4j.gds.core.utils.mem.MemoryRange featureDimension,
 boolean isReduced)