ai.djl.training.dataset

Class RandomSampler

java.lang.Object

ai.djl.training.dataset.RandomSampler

All Implemented Interfaces:

Sampler.SubSampler

public class RandomSampler
extends java.lang.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 and Description
RandomSampler() Creates a new instance of RandomSampler.
RandomSampler(int seed) Creates a new instance of RandomSampler with the given seed.

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

RandomSampler

public RandomSampler()

Creates a new instance of RandomSampler.

RandomSampler

public RandomSampler(int seed)

Creates a new instance of RandomSampler with the given seed.

Parameters:

seed - the value of the seed

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