ai.djl.basicdataset.nlp

Class AmazonReview.Builder

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T>

ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder<AmazonReview.Builder>

ai.djl.basicdataset.nlp.AmazonReview.Builder

Enclosing class:

AmazonReview

public static final class AmazonReview.Builder
extends CsvDataset.CsvBuilder<AmazonReview.Builder>

A builder to construct a AmazonReview.

Field Summary

Fields inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder

csvFormat, csvUrl, features, labels

Fields inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder

dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline

Method Summary

Modifier and Type	Method and Description
AmazonReview	build() Builds the new CsvDataset.
AmazonReview.Builder	optArtifactId(java.lang.String artifactId) Sets the optional artifactId.
AmazonReview.Builder	optDatasetName(java.lang.String datasetName) Sets the name of the subset of Amazon Reviews.
AmazonReview.Builder	optGroupId(java.lang.String groupId) Sets optional groupId.
AmazonReview.Builder	optRepository(ai.djl.repository.Repository repository) Sets the optional repository.
AmazonReview.Builder	self()

Methods inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder

addCategoricalFeature, addCategoricalFeature, addCategoricalLabel, addCategoricalLabel, addFeature, addLabel, addNumericFeature, addNumericLabel, optCsvFile, optCsvUrl, setCsvFormat

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

public AmazonReview.Builder self()

Overrides:

self in class CsvDataset.CsvBuilder<AmazonReview.Builder>

optRepository

public AmazonReview.Builder optRepository(ai.djl.repository.Repository repository)

Sets the optional repository.

Parameters:

repository - the repository

Returns:

this builder

optGroupId

public AmazonReview.Builder optGroupId(java.lang.String groupId)

Sets optional groupId.

Parameters:

groupId - the groupId}

Returns:

this builder

optArtifactId

public AmazonReview.Builder optArtifactId(java.lang.String artifactId)

Sets the optional artifactId.

Parameters:

artifactId - the artifactId

Returns:

this builder

optDatasetName

public AmazonReview.Builder optDatasetName(java.lang.String datasetName)

Sets the name of the subset of Amazon Reviews.

Parameters:

datasetName - the name of the dataset

Returns:

this builder

build

public AmazonReview build()

Builds the new CsvDataset.

Overrides:

build in class CsvDataset.CsvBuilder<AmazonReview.Builder>

Returns:

the new CsvDataset