Package org.neo4j.gds.ml.decisiontree

Class DecisionTreeRegressorTrainer

java.lang.Object

org.neo4j.gds.ml.decisiontree.DecisionTreeTrainer<Double>

org.neo4j.gds.ml.decisiontree.DecisionTreeRegressorTrainer

public class DecisionTreeRegressorTrainer
extends DecisionTreeTrainer<Double>

Constructor Summary

Constructors

Constructor	Description
DecisionTreeRegressorTrainer(ImpurityCriterion impurityCriterion, Features features, org.neo4j.gds.collections.ha.HugeDoubleArray targets, DecisionTreeTrainerConfig config, FeatureBagger featureBagger)

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryRange	memoryEstimation(DecisionTreeTrainerConfig config, long numberOfTrainingSamples)
protected Double	toTerminal(org.neo4j.gds.ml.decisiontree.Group group)

Methods inherited from class org.neo4j.gds.ml.decisiontree.DecisionTreeTrainer

estimateTree, train

Methods inherited from class java.lang.Object

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

Constructor Details

DecisionTreeRegressorTrainer

public DecisionTreeRegressorTrainer(ImpurityCriterion impurityCriterion,
 Features features,
 org.neo4j.gds.collections.ha.HugeDoubleArray targets,
 DecisionTreeTrainerConfig config,
 FeatureBagger featureBagger)

Method Details

memoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryRange memoryEstimation(DecisionTreeTrainerConfig config,
 long numberOfTrainingSamples)

toTerminal

protected Double toTerminal(org.neo4j.gds.ml.decisiontree.Group group)

Specified by:

toTerminal in class DecisionTreeTrainer<Double>