ai.djl.training.dataset

Interface Dataset

All Known Implementing Classes:

ArrayDataset, RandomAccessDataset

public interface Dataset

An interface to represent a set of sample data/label pairs to train a model.

Nested Class Summary

Nested Classes

Modifier and Type	Interface and Description
static class	Dataset.Usage An enum that indicates the mode - training, test or validation.

Method Summary

Modifier and Type	Method and Description
java.lang.Iterable<Batch>	getData(NDManager manager) Fetches an iterator that can iterate through the Dataset.

Method Detail

getData

java.lang.Iterable<Batch> getData(NDManager manager)

Fetches an iterator that can iterate through the Dataset.

Parameters:

manager - the dataset to iterate through

Returns:

an Iterable of Batch that contains batches of data from the dataset