Package org.neo4j.gds.ml.decisiontree
Class DecisionTreeRegressorTrainer
java.lang.Object
org.neo4j.gds.ml.decisiontree.DecisionTreeTrainer<Double>
org.neo4j.gds.ml.decisiontree.DecisionTreeRegressorTrainer
-
Constructor Summary
ConstructorsConstructorDescriptionDecisionTreeRegressorTrainer(ImpurityCriterion impurityCriterion, Features features, org.neo4j.gds.collections.ha.HugeDoubleArray targets, DecisionTreeTrainerConfig config, FeatureBagger featureBagger) -
Method Summary
Modifier and TypeMethodDescriptionstatic org.neo4j.gds.core.utils.mem.MemoryRangememoryEstimation(DecisionTreeTrainerConfig config, long numberOfTrainingSamples) protected DoubletoTerminal(org.neo4j.gds.ml.decisiontree.Group group) Methods inherited from class org.neo4j.gds.ml.decisiontree.DecisionTreeTrainer
estimateTree, train
-
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
- Specified by:
toTerminalin classDecisionTreeTrainer<Double>
-