org.apache.ignite.ml.tree.data

Class DecisionTreeData

java.lang.Object

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

All Implemented Interfaces:

AutoCloseable

public class DecisionTreeData
extends Object
implements AutoCloseable

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

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
void	close()
DecisionTreeData	filter(TreeFilter filter) Filters objects and returns only data that passed filter.
double[][]	getFeatures()
double[]	getLabels()
void	sort(int col) Sorts data by specified column in ascending order.

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)

Constructs a new instance of decision tree data.

Parameters:

features - Matrix with features.

labels - Vector with labels.

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.

getFeatures

public double[][] getFeatures()

getLabels

public double[] getLabels()

close

public void close()

Specified by:

close in interface AutoCloseable