Package ai.djl.basicdataset.cv.classification

Class 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

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      Mnist build()
      Builds the Mnist.
      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()

      • 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 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​(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