org.apache.ignite.ml.tree.data

Class DecisionTreeData

java.lang.Object

org.apache.ignite.ml.dataset.primitive.FeatureMatrixWithLabelsOnHeapData

org.apache.ignite.ml.tree.data.DecisionTreeData

All Implemented Interfaces:

AutoCloseable

public class DecisionTreeData
extends FeatureMatrixWithLabelsOnHeapData
implements AutoCloseable

A partition data of the containing matrix of features and vector of labels stored in heap with index on features.

Constructor Summary

Constructors

Constructor and Description
DecisionTreeData(double[][] features, double[] labels, boolean buildIdx) Constructs a new instance of decision tree data.

Method Summary

Modifier and Type	Method and Description
void	close()
TreeDataIndex	createIndexByFilter(int depth, TreeFilter filter) Builds index in according to current tree depth and cached indexes in upper levels.
DecisionTreeData	filter(TreeFilter filter) Filters objects and returns only data that passed filter.
double[]	getCopiedOriginalLabels()
void	setCopiedOriginalLabels(double[] copiedOriginalLabels)
void	sort(int col) Sorts data by specified column in ascending order.

Methods inherited from class org.apache.ignite.ml.dataset.primitive.FeatureMatrixWithLabelsOnHeapData

getFeatures, getLabels

Methods inherited from class java.lang.Object

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

Constructor Detail

DecisionTreeData

public DecisionTreeData(double[][] features,
                        double[] labels,
                        boolean buildIdx)

Constructs a new instance of decision tree data.

Parameters:

features - Matrix with features.

labels - Vector with labels.

buildIdx - Build index.

Method Detail

filter

public DecisionTreeData filter(TreeFilter filter)

Filters objects and returns only data that passed filter.

Parameters:

filter - Filter.

Returns:

Data passed filter.

sort

public void sort(int col)

Sorts data by specified column in ascending order.

Parameters:

col - Column.

getCopiedOriginalLabels

public double[] getCopiedOriginalLabels()

setCopiedOriginalLabels

public void setCopiedOriginalLabels(double[] copiedOriginalLabels)

close

public void close()

Specified by:

close in interface AutoCloseable

Overrides:

close in class FeatureMatrixWithLabelsOnHeapData

createIndexByFilter

public TreeDataIndex createIndexByFilter(int depth,
                                         TreeFilter filter)

Builds index in according to current tree depth and cached indexes in upper levels. Uses depth as key of cached index and replaces cached index with same key.

Parameters:

depth - Tree Depth.

filter - Filter.