Class WikiText2.Builder

  • Enclosing class:
    WikiText2

    public static final class WikiText2.Builder
    extends java.lang.Object
    A builder to construct a WikiText2 .
    • Method Detail

      • optRepository

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

        public WikiText2.Builder optGroupId​(java.lang.String groupId)
        Sets optional groupId.
        Parameters:
        groupId - the groupId
        Returns:
        this builder
      • optArtifactId

        public WikiText2.Builder optArtifactId​(java.lang.String artifactId)
        Sets the optional artifactId.
        Parameters:
        artifactId - the artifactId
        Returns:
        this builder
      • optUsage

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