org.apache.ignite.ml.selection.split

Class TrainTestSplit<K,V>

java.lang.Object

org.apache.ignite.ml.selection.split.TrainTestSplit<K,V>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

All Implemented Interfaces:

Serializable

public class TrainTestSplit<K,V>
extends Object
implements Serializable

Dataset split that encapsulates train and test subsets.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
TrainTestSplit(IgniteBiPredicate<K,V> trainFilter, IgniteBiPredicate<K,V> testFilter) Constructs a new instance of train test split.

Method Summary

Modifier and Type	Method and Description
IgniteBiPredicate<K,V>	getTestFilter()
IgniteBiPredicate<K,V>	getTrainFilter()

Methods inherited from class java.lang.Object

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

Constructor Detail

TrainTestSplit

public TrainTestSplit(IgniteBiPredicate<K,V> trainFilter,
                      IgniteBiPredicate<K,V> testFilter)

Constructs a new instance of train test split.

Parameters:

trainFilter - Filter that passes train subset of the dataset.

testFilter - Filter that passes test subset of the dataset.

Method Detail

getTrainFilter

public IgniteBiPredicate<K,V> getTrainFilter()

getTestFilter

public IgniteBiPredicate<K,V> getTestFilter()