Package 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 anArrayDataset.
-
-
Field Summary
-
Fields inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ArrayDatasetbuild()Builds a new instance ofArrayDatasetwith the specified data and labels.ArrayDataset.BuilderoptLabels(NDArray... labels)Sets the labels for the data in theArrayDataset.protected ArrayDataset.Builderself()Returns this {code Builder} object.ArrayDataset.BuildersetData(NDArray... data)Sets the data as anNDArrayfor theArrayDataset.-
Methods inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling
-
-
-
-
Method Detail
-
self
protected ArrayDataset.Builder self()
Returns this {code Builder} object.- Specified by:
selfin classRandomAccessDataset.BaseBuilder<ArrayDataset.Builder>- Returns:
- this
BaseBuilder
-
setData
public ArrayDataset.Builder setData(NDArray... data)
Sets the data as anNDArrayfor theArrayDataset.- Parameters:
data- an array ofNDArraythat contains the data- Returns:
- this Builder
-
optLabels
public ArrayDataset.Builder optLabels(NDArray... labels)
Sets the labels for the data in theArrayDataset.- Parameters:
labels- an array ofNDArraythat contains the labels- Returns:
- this Builder
-
build
public ArrayDataset build()
Builds a new instance ofArrayDatasetwith the specified data and labels.- Returns:
- a new instance of
ArrayDataset
-
-