org.apache.ignite.ml.tree

Class DecisionTreeRegressionTrainer

java.lang.Object

org.apache.ignite.ml.tree.DecisionTreeRegressionTrainer

All Implemented Interfaces:

DatasetTrainer<DecisionTreeNode,Double>

public class DecisionTreeRegressionTrainer
extends Object

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

Constructor Summary

Constructors

Constructor and Description
DecisionTreeRegressionTrainer(int maxDeep, double minImpurityDecrease) Constructs a new decision tree regressor with default impurity function compressor.
DecisionTreeRegressionTrainer(int maxDeep, double minImpurityDecrease, StepFunctionCompressor<MSEImpurityMeasure> compressor) Constructs a new decision tree regressor.

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

DecisionTreeRegressionTrainer

public DecisionTreeRegressionTrainer(int maxDeep,
                                     double minImpurityDecrease)

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

Parameters:

maxDeep - Max tree deep.

minImpurityDecrease - Min impurity decrease.

DecisionTreeRegressionTrainer

public DecisionTreeRegressionTrainer(int maxDeep,
                                     double minImpurityDecrease,
                                     StepFunctionCompressor<MSEImpurityMeasure> compressor)

Constructs a new decision tree regressor.

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.