org.apache.ignite.ml.tree

Class DecisionTreeClassificationTrainer

java.lang.Object

org.apache.ignite.ml.trainers.DatasetTrainer<DecisionTreeNode,Double>

org.apache.ignite.ml.tree.DecisionTree<GiniImpurityMeasure>

org.apache.ignite.ml.tree.DecisionTreeClassificationTrainer

public class DecisionTreeClassificationTrainer
extends DecisionTree<GiniImpurityMeasure>

Decision tree classifier based on distributed decision tree trainer that allows to fit trees using row-partitioned dataset.

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

DatasetTrainer.EmptyDatasetException

Field Summary

Fields inherited from class org.apache.ignite.ml.tree.DecisionTree

usingIdx

Fields inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

environment

Constructor Summary

Constructors

Constructor and Description
DecisionTreeClassificationTrainer() Constructs a new decision tree classifier with default impurity function compressor and default maxDeep = 5 and minImpurityDecrease = 0.
DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease) Constructs a new decision tree classifier with default impurity function compressor.
DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease, StepFunctionCompressor<GiniImpurityMeasure> compressor) Constructs a new instance of decision tree classifier.

Method Summary

Modifier and Type	Method and Description
protected ImpurityMeasureCalculator<GiniImpurityMeasure>	getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset) Returns impurity measure calculator.
DecisionTreeClassificationTrainer	withMaxDeep(Double maxDeep) Set up the max deep of decision tree.
DecisionTreeClassificationTrainer	withMinImpurityDecrease(Double minImpurityDecrease) Set up the min impurity decrease of decision tree.
DecisionTreeClassificationTrainer	withUseIndex(boolean useIdx) Sets useIndex parameter and returns trainer instance.

Methods inherited from class org.apache.ignite.ml.tree.DecisionTree

checkState, fit, fit, printTree, updateModel

Methods inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

fit, fit, fit, fit, getLastTrainedModelOrThrowEmptyDatasetException, setEnvironment, update, update, update, update, update

Methods inherited from class java.lang.Object

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

Constructor Detail

DecisionTreeClassificationTrainer

public DecisionTreeClassificationTrainer(int maxDeep,
                                         double minImpurityDecrease)

Constructs a new decision tree classifier with default impurity function compressor.

Parameters:

maxDeep - Max tree deep.

minImpurityDecrease - Min impurity decrease.

DecisionTreeClassificationTrainer

public DecisionTreeClassificationTrainer()

Constructs a new decision tree classifier with default impurity function compressor and default maxDeep = 5 and minImpurityDecrease = 0.

DecisionTreeClassificationTrainer

public DecisionTreeClassificationTrainer(int maxDeep,
                                         double minImpurityDecrease,
                                         StepFunctionCompressor<GiniImpurityMeasure> compressor)

Constructs a new instance of decision tree classifier.

Parameters:

maxDeep - Max tree deep.

minImpurityDecrease - Min impurity decrease.

Method Detail

withMaxDeep

public DecisionTreeClassificationTrainer withMaxDeep(Double maxDeep)

Set up the max deep of decision tree.

Parameters:

maxDeep - The parameter value.

Returns:

Trainer with new maxDeep parameter value.

withMinImpurityDecrease

public DecisionTreeClassificationTrainer withMinImpurityDecrease(Double minImpurityDecrease)

Set up the min impurity decrease of decision tree.

Parameters:

minImpurityDecrease - The parameter value.

Returns:

Trainer with new minImpurityDecrease parameter value.

withUseIndex

public DecisionTreeClassificationTrainer withUseIndex(boolean useIdx)

Sets useIndex parameter and returns trainer instance.

Parameters:

useIdx - Use index.

Returns:

Decision tree trainer.

getImpurityMeasureCalculator

protected ImpurityMeasureCalculator<GiniImpurityMeasure> getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)

Returns impurity measure calculator.

Specified by:

getImpurityMeasureCalculator in class DecisionTree<GiniImpurityMeasure>

Parameters:

dataset - Dataset.

Returns:

Impurity measure calculator.