Package ai.djl.training.dataset

Class SequenceSampler

java.lang.Object

ai.djl.training.dataset.SequenceSampler

All Implemented Interfaces:

Sampler.SubSampler

public class SequenceSampler
extends Object
implements Sampler.SubSampler

SequenceSampler is an implementation of the Sampler.SubSampler interface.

SequenceSampler samples the data from [0, dataset.size) sequentially.

Constructor Summary

Constructors

Constructor	Description
SequenceSampler()

Method Summary

Modifier and Type	Method	Description
Iterator<Long>	sample(RandomAccessDataset dataset)	Fetches an iterator that iterates through the indices of the given RandomAccessDataset.

Methods inherited from class java.lang.Object

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

Constructor Details

SequenceSampler

public SequenceSampler()

Method Details

sample

public Iterator<Long> sample(RandomAccessDataset dataset)

Fetches an iterator that iterates through the indices of the given RandomAccessDataset.

Specified by:

sample in interface Sampler.SubSampler

Parameters:

dataset - the RandomAccessDataset to sample from

Returns:

an iterator that iterates through the indices of the given RandomAccessDataset