org.apache.ignite.ml.tree

Class DecisionTreeClassificationTrainer

java.lang.Object

org.apache.ignite.ml.tree.DecisionTreeClassificationTrainer

All Implemented Interfaces:

DatasetTrainer<DecisionTreeNode,Double>

public class DecisionTreeClassificationTrainer
extends Object

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

Constructor Summary

Constructors

Constructor and Description
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
<K,V> DecisionTreeNode	fit(DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,double[]> featureExtractor, IgniteBiFunction<K,V,Double> lbExtractor) Trains model based on the specified data.

Methods inherited from class java.lang.Object

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

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

fit, fit

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(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

fit

public <K,V> DecisionTreeNode fit(DatasetBuilder<K,V> datasetBuilder,
                                  IgniteBiFunction<K,V,double[]> featureExtractor,
                                  IgniteBiFunction<K,V,Double> lbExtractor)

Trains model based on the specified data.

Specified by:

fit in interface DatasetTrainer<DecisionTreeNode,Double>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

Parameters:

datasetBuilder - Dataset builder.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Model.