Package org.neo4j.gds.ml.decisiontree
Class DecisionTreeClassifierTrainer
java.lang.Object
org.neo4j.gds.ml.decisiontree.DecisionTreeTrainer<Integer>
org.neo4j.gds.ml.decisiontree.DecisionTreeClassifierTrainer
-
Constructor Summary
ConstructorsConstructorDescriptionDecisionTreeClassifierTrainer(ImpurityCriterion impurityCriterion, Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, int numberOfClasses, DecisionTreeTrainerConfig config, FeatureBagger featureBagger) -
Method Summary
Modifier and TypeMethodDescriptionstatic org.neo4j.gds.core.utils.mem.MemoryRangememoryEstimation(DecisionTreeTrainerConfig config, long numberOfTrainingSamples, int numberOfClasses) protected IntegertoTerminal(org.neo4j.gds.ml.decisiontree.Group group) Methods inherited from class org.neo4j.gds.ml.decisiontree.DecisionTreeTrainer
estimateTree, train
-
Constructor Details
-
DecisionTreeClassifierTrainer
public DecisionTreeClassifierTrainer(ImpurityCriterion impurityCriterion, Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, int numberOfClasses, DecisionTreeTrainerConfig config, FeatureBagger featureBagger)
-
-
Method Details
-
memoryEstimation
public static org.neo4j.gds.core.utils.mem.MemoryRange memoryEstimation(DecisionTreeTrainerConfig config, long numberOfTrainingSamples, int numberOfClasses) -
toTerminal
- Specified by:
toTerminalin classDecisionTreeTrainer<Integer>
-