Package org.neo4j.gds.ml.decisiontree

Class DecisionTreeTrainer<PREDICTION extends Number>

java.lang.Object

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

Direct Known Subclasses:

DecisionTreeClassifierTrainer, DecisionTreeRegressorTrainer

public abstract class DecisionTreeTrainer<PREDICTION extends Number>
extends Object

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryRange	estimateTree(DecisionTreeTrainerConfig config, long numberOfTrainingSamples, long leafNodeSizeInBytes)
protected abstract PREDICTION	toTerminal(org.neo4j.gds.ml.decisiontree.Group group)
DecisionTreePredictor<PREDICTION>	train(org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSetIndices)

Methods inherited from class java.lang.Object

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

Method Details

estimateTree

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

train

public DecisionTreePredictor<PREDICTION> train(org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSetIndices)

toTerminal

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