Package ai.djl.basicdataset.cv.classification

Class Cifar10.Builder

java.lang.Object
ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Cifar10.Builder>
ai.djl.basicdataset.cv.classification.Cifar10.Builder
Enclosing class:
Cifar10
public static final class Cifar10.Builder extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Cifar10.Builder>
A builder to construct a Cifar10.

  • 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
    Cifar10
    build()
    Builds a new Cifar10.
    Cifar10.Builder
    optArtifactId(String artifactId)
    Sets the optional artifactId.
    Cifar10.Builder
    optGroupId(String groupId)
    Sets optional groupId.
    Cifar10.Builder
    optManager(ai.djl.ndarray.NDManager manager)
    Sets the optional manager for the dataset (default follows engine default).
    Cifar10.Builder
    optRepository(ai.djl.repository.Repository repository)
    Sets the optional repository for the dataset.
    Cifar10.Builder
    optUsage(ai.djl.training.dataset.Dataset.Usage usage)
    Sets the optional usage for the dataset.
    protected Cifar10.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 Cifar10.Builder self()
      Specified by:
      self in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Cifar10.Builder>

    • optManager

      public Cifar10.Builder optManager(ai.djl.ndarray.NDManager manager)
      Sets the optional manager for the dataset (default follows engine default).
      Parameters:
      manager - the new manager
      Returns:
      this builder

    • optRepository

      public Cifar10.Builder optRepository(ai.djl.repository.Repository repository)
      Sets the optional repository for the dataset.
      Parameters:
      repository - the new repository
      Returns:
      this builder

    • optGroupId

      public Cifar10.Builder optGroupId(String groupId)
      Sets optional groupId.
      Parameters:
      groupId - the groupId}
      Returns:
      this builder

    • optArtifactId

      public Cifar10.Builder optArtifactId(String artifactId)
      Sets the optional artifactId.
      Parameters:
      artifactId - the artifactId
      Returns:
      this builder

    • optUsage

      public Cifar10.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)
      Sets the optional usage for the dataset.
      Parameters:
      usage - the usage
      Returns:
      this builder

    • build

      public Cifar10 build()
      Builds a new Cifar10.
      Returns:
      the new Cifar10