org.apache.ignite.ml.tree.leaf

Class MeanDecisionTreeLeafBuilder

java.lang.Object

org.apache.ignite.ml.tree.leaf.MeanDecisionTreeLeafBuilder

All Implemented Interfaces:

DecisionTreeLeafBuilder

public class MeanDecisionTreeLeafBuilder
extends Object
implements DecisionTreeLeafBuilder

Decision tree leaf node builder that chooses mean value as a leaf value.

Constructor Summary

Constructors

Constructor and Description
MeanDecisionTreeLeafBuilder()

Method Summary

Modifier and Type	Method and Description
DecisionTreeLeafNode	createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset, TreeFilter pred) Creates new leaf node for given dataset and node predicate.

Methods inherited from class java.lang.Object

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

Constructor Detail

MeanDecisionTreeLeafBuilder

public MeanDecisionTreeLeafBuilder()

Method Detail

createLeafNode

public DecisionTreeLeafNode createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset,
                                           TreeFilter pred)

Creates new leaf node for given dataset and node predicate.

Specified by:

createLeafNode in interface DecisionTreeLeafBuilder

Parameters:

dataset - Dataset.

pred - Node predicate.

Returns:

Leaf node.