ai.djl.training.dataset

Class ArrayDataset.Builder

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<ArrayDataset.Builder>

ai.djl.training.dataset.ArrayDataset.Builder

Enclosing class:

ArrayDataset

public static final class ArrayDataset.Builder
extends RandomAccessDataset.BaseBuilder<ArrayDataset.Builder>

The Builder to construct an ArrayDataset.

Field Summary

Fields inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder

dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
ArrayDataset	build() Builds a new instance of ArrayDataset with the specified data and labels.
ArrayDataset.Builder	optLabels(NDArray... labels) Sets the labels for the data in the ArrayDataset.
protected ArrayDataset.Builder	self() Returns this {code Builder} object.
ArrayDataset.Builder	setData(NDArray... data) Sets the data as an NDArray for the ArrayDataset.

Methods inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder

addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

self

protected ArrayDataset.Builder self()

Returns this {code Builder} object.

Specified by:

self in class RandomAccessDataset.BaseBuilder<ArrayDataset.Builder>

Returns:

this BaseBuilder

setData

public ArrayDataset.Builder setData(NDArray... data)

Sets the data as an NDArray for the ArrayDataset.

Parameters:

data - an array of NDArray that contains the data

Returns:

this Builder

optLabels

public ArrayDataset.Builder optLabels(NDArray... labels)

Sets the labels for the data in the ArrayDataset.

Parameters:

labels - an array of NDArray that contains the labels

Returns:

this Builder

build

public ArrayDataset build()

Builds a new instance of ArrayDataset with the specified data and labels.

Returns:

a new instance of ArrayDataset