org.apache.ignite.ml.structures

Class LabeledDatasetTestTrainPair

java.lang.Object

org.apache.ignite.ml.structures.LabeledDatasetTestTrainPair

All Implemented Interfaces:

Serializable

public class LabeledDatasetTestTrainPair
extends Object
implements Serializable

Class for splitting Labeled Dataset on train and test sets.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
LabeledDatasetTestTrainPair(LabeledDataset dataset, double testPercentage) Creates two subsets of given dataset.

Method Summary

Modifier and Type	Method and Description
LabeledDataset	test() Test subset of the whole dataset.
LabeledDataset	train() Train subset of the whole dataset.

Methods inherited from class java.lang.Object

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

Constructor Detail

LabeledDatasetTestTrainPair

public LabeledDatasetTestTrainPair(LabeledDataset dataset,
                                   double testPercentage)

Creates two subsets of given dataset.

NOTE: This method uses next algorithm with O(n log n) by calculations and O(n) by memory.

Parameters:

dataset - The dataset to split on train and test subsets.

testPercentage - The percentage of the test subset.

Method Detail

train

public LabeledDataset train()

Train subset of the whole dataset.

Returns:

Train subset.

test

public LabeledDataset test()

Test subset of the whole dataset.

Returns:

Test subset.