edu.umass.cs.mallet.base.types

Class InstanceList.CrossValidationIterator

java.lang.Object

edu.umass.cs.mallet.base.types.InstanceList.CrossValidationIterator

All Implemented Interfaces:

Serializable, Iterator

Enclosing class:

InstanceList

public class InstanceList.CrossValidationIterator
extends Object
implements Iterator, Serializable

CrossValidationIterator allows iterating over pairs of InstanceList, where each pair is split into training/testing based on nfolds.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
CrossValidationIterator(int _nfolds)
CrossValidationIterator(int _nfolds, int seed)

Method Summary

Modifier and Type	Method and Description
boolean	hasNext()
Object	next()
InstanceList[]	nextSplit() Returns the next training/testing split.
InstanceList[]	nextSplit(int numTrainFolds) Returns the next split, given the number of folds you want in the training data.
void	remove()

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.Iterator

forEachRemaining

Constructor Detail

CrossValidationIterator

public CrossValidationIterator(int _nfolds,
                               int seed)

Parameters:

_nfolds - number of folds to split InstanceList into

seed - seed for random number used to split InstanceList

CrossValidationIterator

public CrossValidationIterator(int _nfolds)

Method Detail

hasNext

public boolean hasNext()

Specified by:

hasNext in interface Iterator

nextSplit

public InstanceList[] nextSplit()

Returns the next training/testing split.

Returns:

A pair of lists, where InstanceList[0] is the larger split (training) and InstanceList[1] is the smaller split (testing)

nextSplit

public InstanceList[] nextSplit(int numTrainFolds)

Returns the next split, given the number of folds you want in the training data.

next

public Object next()

Specified by:

next in interface Iterator

remove

public void remove()

Specified by:

remove in interface Iterator