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

Class RandomForestRegressorTrainer

java.lang.Object

org.neo4j.gds.ml.models.randomforest.RandomForestRegressorTrainer

All Implemented Interfaces:

RegressorTrainer

public class RandomForestRegressorTrainer
extends Object
implements RegressorTrainer

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(LongUnaryOperator numberOfTrainingSamples, org.neo4j.gds.core.utils.mem.MemoryRange featureDimension, RandomForestRegressorTrainerConfig config)
RandomForestRegressor	train(Features allFeatureVectors, org.neo4j.gds.collections.ha.HugeDoubleArray targets, 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

RandomForestRegressorTrainer

public RandomForestRegressorTrainer(int concurrency,
 RandomForestRegressorTrainerConfig config,
 Optional<Long> randomSeed,
 org.neo4j.gds.termination.TerminationFlag terminationFlag,
 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
 org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel)

Method Details

memoryEstimation

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

train

public RandomForestRegressor train(Features allFeatureVectors,
 org.neo4j.gds.collections.ha.HugeDoubleArray targets,
 org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)

Specified by:

train in interface RegressorTrainer