Package org.neo4j.gds.ml.models.randomforest

Class RandomForestClassifierTrainer

java.lang.Object

org.neo4j.gds.ml.models.randomforest.RandomForestClassifierTrainer

All Implemented Interfaces:

ClassifierTrainer

public class RandomForestClassifierTrainer
extends Object
implements ClassifierTrainer

Constructor Summary

Constructors

Constructor	Description
RandomForestClassifierTrainer(int concurrency, int numberOfClasses, RandomForestClassifierTrainerConfig config, Optional<Long> randomSeed, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel, org.neo4j.gds.termination.TerminationFlag terminationFlag, ModelSpecificMetricsHandler metricsHandler)

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(LongUnaryOperator numberOfTrainingSamples, int numberOfClasses, org.neo4j.gds.core.utils.mem.MemoryRange featureDimension, RandomForestClassifierTrainerConfig config)
RandomForestClassifier	train(Features allFeatureVectors, org.neo4j.gds.collections.ha.HugeIntArray allLabels, org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)

Methods inherited from class java.lang.Object

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

Constructor Details

RandomForestClassifierTrainer

public RandomForestClassifierTrainer(int concurrency,
 int numberOfClasses,
 RandomForestClassifierTrainerConfig config,
 Optional<Long> randomSeed,
 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
 org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel,
 org.neo4j.gds.termination.TerminationFlag terminationFlag,
 ModelSpecificMetricsHandler metricsHandler)

Method Details

memoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(LongUnaryOperator numberOfTrainingSamples,
 int numberOfClasses,
 org.neo4j.gds.core.utils.mem.MemoryRange featureDimension,
 RandomForestClassifierTrainerConfig config)

train

public RandomForestClassifier train(Features allFeatureVectors,
 org.neo4j.gds.collections.ha.HugeIntArray allLabels,
 org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)

Specified by:

train in interface ClassifierTrainer