ai.djl.training.dataset

Class SequenceSampler

java.lang.Object

ai.djl.training.dataset.SequenceSampler

All Implemented Interfaces:

Sampler.SubSampler

public class SequenceSampler
extends java.lang.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 and Description
SequenceSampler()

Method Summary

Modifier and Type	Method and Description
java.util.Iterator<java.lang.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 Detail

SequenceSampler

public SequenceSampler()

Method Detail

sample

public java.util.Iterator<java.lang.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