Class RandomForestClassifierTrainer
java.lang.Object
org.neo4j.gds.ml.models.randomforest.RandomForestClassifierTrainer
- All Implemented Interfaces:
ClassifierTrainer
-
Constructor Summary
ConstructorsConstructorDescriptionRandomForestClassifierTrainer(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 TypeMethodDescriptionstatic org.neo4j.gds.core.utils.mem.MemoryEstimationmemoryEstimation(LongUnaryOperator numberOfTrainingSamples, int numberOfClasses, org.neo4j.gds.core.utils.mem.MemoryRange featureDimension, RandomForestClassifierTrainerConfig config) train(Features allFeatureVectors, org.neo4j.gds.collections.ha.HugeIntArray allLabels, org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)
-
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:
trainin interfaceClassifierTrainer
-