Package ai.djl.basicdataset.nlp
Class WikiText2.Builder
- java.lang.Object
-
- ai.djl.basicdataset.nlp.WikiText2.Builder
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description WikiText2build()Builds a newWikiText2object.WikiText2.BuilderoptArtifactId(java.lang.String artifactId)Sets the optional artifactId.WikiText2.BuilderoptGroupId(java.lang.String groupId)Sets optional groupId.WikiText2.BuilderoptRepository(ai.djl.repository.Repository repository)Sets the optional repository for the dataset.WikiText2.BuilderoptUsage(ai.djl.training.dataset.Dataset.Usage usage)Sets the optional usage for the dataset.
-
-
-
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
-
-