Package ai.djl.basicdataset.nlp
Class AmazonReview.Builder
java.lang.Object
ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T>
ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder<T>
ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder<AmazonReview.Builder>
ai.djl.basicdataset.nlp.AmazonReview.Builder
- Enclosing class:
- AmazonReview
A builder to construct a
AmazonReview.-
Field Summary
Fields inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
csvFormat, csvUrlFields inherited from class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
allowNoLabels, features, labelsFields inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline -
Method Summary
Modifier and TypeMethodDescriptionbuild()Builds the newCsvDataset.optArtifactId(String artifactId) Sets the optional artifactId.optDatasetName(String datasetName) Sets the name of the subset of Amazon Reviews.optGroupId(String groupId) Sets optional groupId.optRepository(ai.djl.repository.Repository repository) Sets the optional repository.self()Methods inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
optCsvFile, optCsvUrl, setCsvFormatMethods inherited from class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
addCategoricalFeature, addCategoricalFeature, addCategoricalFeature, addCategoricalLabel, addCategoricalLabel, addCategoricalLabel, addFeature, addLabel, addNumericFeature, addNumericFeature, addNumericLabel, addNumericLabel, noLabelsMethods inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling
-
Method Details
-
self
- Overrides:
selfin classCsvDataset.CsvBuilder<AmazonReview.Builder>
-
optRepository
Sets the optional repository.- Parameters:
repository- the repository- Returns:
- this builder
-
optGroupId
Sets optional groupId.- Parameters:
groupId- the groupId}- Returns:
- this builder
-
optArtifactId
Sets the optional artifactId.- Parameters:
artifactId- the artifactId- Returns:
- this builder
-
optDatasetName
Sets the name of the subset of Amazon Reviews.- Parameters:
datasetName- the name of the dataset- Returns:
- this builder
-
build
Builds the newCsvDataset.- Overrides:
buildin classCsvDataset.CsvBuilder<AmazonReview.Builder>- Returns:
- the new
CsvDataset
-