ai.djl.basicdataset.cv.classification

Class FashionMnist.Builder

java.lang.Object

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

ai.djl.basicdataset.cv.classification.FashionMnist.Builder

Enclosing class:

FashionMnist

public static final class FashionMnist.Builder
extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<FashionMnist.Builder>

A builder for a FashionMnist.

Field Summary

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

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

Method Summary

Modifier and Type	Method and Description
FashionMnist	build() Builds the Mnist.
FashionMnist.Builder	optArtifactId(java.lang.String artifactId) Sets the optional artifactId.
FashionMnist.Builder	optGroupId(java.lang.String groupId) Sets optional groupId.
FashionMnist.Builder	optManager(ai.djl.ndarray.NDManager manager) Sets the optional manager for the dataset (default follows engine default).
FashionMnist.Builder	optRepository(ai.djl.repository.Repository repository) Sets the optional repository.
FashionMnist.Builder	optUsage(ai.djl.training.dataset.Dataset.Usage usage) Sets the optional usage.
protected FashionMnist.Builder	self()

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

Method Detail

self

protected FashionMnist.Builder self()

Specified by:

self in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<FashionMnist.Builder>

optManager

public FashionMnist.Builder optManager(ai.djl.ndarray.NDManager manager)

Sets the optional manager for the dataset (default follows engine default).

Parameters:

manager - the manager

Returns:

this builder

optRepository

public FashionMnist.Builder optRepository(ai.djl.repository.Repository repository)

Sets the optional repository.

Parameters:

repository - the repository

Returns:

this builder

optGroupId

public FashionMnist.Builder optGroupId(java.lang.String groupId)

Sets optional groupId.

Parameters:

groupId - the groupId}

Returns:

this builder

optArtifactId

public FashionMnist.Builder optArtifactId(java.lang.String artifactId)

Sets the optional artifactId.

Parameters:

artifactId - the artifactId

Returns:

this builder

optUsage

public FashionMnist.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)

Sets the optional usage.

Parameters:

usage - the usage

Returns:

this builder

build

public FashionMnist build()

Builds the Mnist.

Returns:

the Mnist