Package ai.djl.basicdataset.cv.classification

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 a Mnist.

  • 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
    Description
    Mnist
    build()
    Builds the Mnist.
    Mnist.Builder
    optArtifactId(String artifactId)
    Sets the optional artifactId.
    Mnist.Builder
    optGroupId(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()

    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 Details

    • self

      protected Mnist.Builder self()
      Specified by:
      self in class ai.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(String groupId)
      Sets optional groupId.
      Parameters:
      groupId - the groupId}
      Returns:
      this builder

    • optArtifactId

      public Mnist.Builder optArtifactId(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

    • build

      public Mnist build()
      Builds the Mnist.
      Returns:
      the Mnist