Package de.uni_mannheim.informatik.dws.melt.matching_ml.util

Class TrainTestSplit<T>

java.lang.Object

de.uni_mannheim.informatik.dws.melt.matching_ml.util.TrainTestSplit<T>

public class TrainTestSplit<T>
extends Object

A class which can do a train test split for arbitrary data items. It also works with stratification.

Field Summary

Fields

Modifier and Type	Field	Description
private List<T>	test
private List<T>	train

Constructor Summary

Constructors

Constructor	Description
TrainTestSplit(Collection<T> items, double train_ratio)
TrainTestSplit(Collection<T> items, double train_ratio, long seed)
TrainTestSplit(Collection<T> items, double train_ratio, long seed, Function<T,?> stratify)
TrainTestSplit(Collection<T> items, double train_ratio, Function<T,?> stratify)
TrainTestSplit(Collection<T> items, double train_ratio, Random rnd)
TrainTestSplit(Collection<T> items, double train_ratio, Random rnd, Function<T,?> stratify)
TrainTestSplit(List<T> items, double train_ratio, long seed, List<?> stratify)
TrainTestSplit(List<T> items, double train_ratio, List<?> stratify)
TrainTestSplit(List<T> items, double train_ratio, Random rnd, List<?> stratify)

Method Summary

Modifier and Type	Method	Description
List<T>	getTest()	Returns a list of objects in the test set
List<T>	getTrain()	Returns a list of objects in the train set
private void	initializeTrainTestStratified(Map<Object,List<T>> groups, double train_ratio, Random rnd)

Methods inherited from class java.lang.Object

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

Field Details

train

private List<T> train

test

private List<T> test

Constructor Details

TrainTestSplit

public TrainTestSplit(List<T> items,
 double train_ratio,
 Random rnd,
 List<?> stratify)

TrainTestSplit

public TrainTestSplit(List<T> items,
 double train_ratio,
 long seed,
 List<?> stratify)

TrainTestSplit

public TrainTestSplit(List<T> items,
 double train_ratio,
 List<?> stratify)

TrainTestSplit

public TrainTestSplit(Collection<T> items,
 double train_ratio,
 Random rnd,
 Function<T,?> stratify)

TrainTestSplit

public TrainTestSplit(Collection<T> items,
 double train_ratio,
 long seed,
 Function<T,?> stratify)

TrainTestSplit

public TrainTestSplit(Collection<T> items,
 double train_ratio,
 Function<T,?> stratify)

TrainTestSplit

public TrainTestSplit(Collection<T> items,
 double train_ratio,
 Random rnd)

TrainTestSplit

public TrainTestSplit(Collection<T> items,
 double train_ratio,
 long seed)

TrainTestSplit

public TrainTestSplit(Collection<T> items,
 double train_ratio)

Method Details

initializeTrainTestStratified

private void initializeTrainTestStratified(Map<Object,List<T>> groups,
 double train_ratio,
 Random rnd)

getTrain

public List<T> getTrain()

Returns a list of objects in the train set

Returns:

the train list

getTest

public List<T> getTest()

Returns a list of objects in the test set

Returns:

the test list