Class Mnist.Builder
- java.lang.Object
-
- ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Mnist.Builder>
-
- ai.djl.basicdataset.cv.classification.Mnist.Builder
-
- Enclosing class:
- Mnist
public static final class Mnist.Builder extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Mnist.Builder>
A builder for aMnist
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Mnist
build()
Builds theMnist
.Mnist.Builder
optArtifactId(java.lang.String artifactId)
Sets the optional artifactId.Mnist.Builder
optGroupId(java.lang.String groupId)
Sets optional groupId.Mnist.Builder
optManager(ai.djl.ndarray.NDManager manager)
Sets the optional manager for the dataset (default follows engine default).Mnist.Builder
optRepository(ai.djl.repository.Repository repository)
Sets the optional repository.Mnist.Builder
optUsage(ai.djl.training.dataset.Dataset.Usage usage)
Sets the optional usage.protected Mnist.Builder
self()
-
-
-
Method Detail
-
self
protected Mnist.Builder self()
- Specified by:
self
in classai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Mnist.Builder>
-
optManager
public Mnist.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 Mnist.Builder optRepository(ai.djl.repository.Repository repository)
Sets the optional repository.- Parameters:
repository
- the repository- Returns:
- this builder
-
optGroupId
public Mnist.Builder optGroupId(java.lang.String groupId)
Sets optional groupId.- Parameters:
groupId
- the groupId}- Returns:
- this builder
-
optArtifactId
public Mnist.Builder optArtifactId(java.lang.String artifactId)
Sets the optional artifactId.- Parameters:
artifactId
- the artifactId- Returns:
- this builder
-
optUsage
public Mnist.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)
Sets the optional usage.- Parameters:
usage
- the usage- Returns:
- this builder
-
-