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 ArrayDataset
build()
Builds a new instance ofArrayDataset
with the specified data and labels.ArrayDataset.Builder
optLabels(NDArray... labels)
Sets the labels for the data in theArrayDataset
.protected ArrayDataset.Builder
self()
Returns this {code Builder} object.ArrayDataset.Builder
setData(NDArray... data)
Sets the data as anNDArray
for 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:
self
in classRandomAccessDataset.BaseBuilder<ArrayDataset.Builder>
- Returns:
- this
BaseBuilder
-
setData
public ArrayDataset.Builder setData(NDArray... data)
Sets the data as anNDArray
for theArrayDataset
.- Parameters:
data
- an array ofNDArray
that 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 ofNDArray
that contains the labels- Returns:
- this Builder
-
build
public ArrayDataset build()
Builds a new instance ofArrayDataset
with the specified data and labels.- Returns:
- a new instance of
ArrayDataset
-
-