Package ai.djl.training.dataset

Class RandomSampler

java.lang.Object

ai.djl.training.dataset.RandomSampler

All Implemented Interfaces:

Sampler.SubSampler

public class RandomSampler
extends Object
implements Sampler.SubSampler

RandomSampler is an implementation of the Sampler.SubSampler interface.

RandomSampler samples the data from [0, dataset.size) randomly.

Constructor Summary

Constructors

Constructor	Description
RandomSampler()

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

RandomSampler

public RandomSampler()

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