Package org.neo4j.gds.ml.decisiontree

Class DecisionTreeClassifierTrainer

java.lang.Object

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

org.neo4j.gds.ml.decisiontree.DecisionTreeClassifierTrainer

public class DecisionTreeClassifierTrainer
extends DecisionTreeTrainer<Integer>

Constructor Summary

Constructors

Constructor	Description
DecisionTreeClassifierTrainer(ImpurityCriterion impurityCriterion, Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, int numberOfClasses, DecisionTreeTrainerConfig config, FeatureBagger featureBagger)

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryRange	memoryEstimation(DecisionTreeTrainerConfig config, long numberOfTrainingSamples, int numberOfClasses)
protected Integer	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

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

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

Specified by:

toTerminal in class DecisionTreeTrainer<Integer>