Package ai.djl.basicdataset.nlp
Class AmazonReview
java.lang.Object
ai.djl.training.dataset.RandomAccessDataset
ai.djl.basicdataset.tabular.TabularDataset
ai.djl.basicdataset.tabular.CsvDataset
ai.djl.basicdataset.nlp.AmazonReview
- All Implemented Interfaces:
ai.djl.training.dataset.Dataset
The
AmazonReview
dataset contains a Application.NLP.SENTIMENT_ANALYSIS
set
of reviews and their sentiment ratings.-
Nested Class Summary
Nested ClassesModifier and TypeClassDescriptionstatic final class
A builder to construct aAmazonReview
.Nested classes/interfaces inherited from class ai.djl.basicdataset.tabular.CsvDataset
CsvDataset.CsvBuilder<T extends CsvDataset.CsvBuilder<T>>
Nested classes/interfaces inherited from class ai.djl.basicdataset.tabular.TabularDataset
TabularDataset.BaseBuilder<T extends TabularDataset.BaseBuilder<T>>
Nested classes/interfaces inherited from interface ai.djl.training.dataset.Dataset
ai.djl.training.dataset.Dataset.Usage
-
Field Summary
Fields inherited from class ai.djl.basicdataset.tabular.CsvDataset
csvFormat, csvRecords, csvUrl
Fields inherited from class ai.djl.basicdataset.tabular.TabularDataset
features, labels
Fields inherited from class ai.djl.training.dataset.RandomAccessDataset
dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline
-
Constructor Summary
ConstructorsModifierConstructorDescriptionprotected
AmazonReview
(AmazonReview.Builder builder) Creates a new instance ofAmazonReview
with the given necessary configurations. -
Method Summary
Modifier and TypeMethodDescriptionstatic AmazonReview.Builder
builder()
Creates a new builder to build aAmazonReview
.void
prepare
(ai.djl.util.Progress progress) Methods inherited from class ai.djl.basicdataset.tabular.CsvDataset
availableSize, getCell, getColumnNames
Methods inherited from class ai.djl.basicdataset.tabular.TabularDataset
get, getFeatures, getFeatureSize, getLabels, getLabelSize, getRowDirect, getRowFeatures, matchingTranslatorOptions, prepareFeaturizers
Methods inherited from class ai.djl.training.dataset.RandomAccessDataset
getData, getData, getData, getData, newSubDataset, newSubDataset, randomSplit, size, subDataset, subDataset, subDataset, subDataset, toArray
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
Methods inherited from interface ai.djl.training.dataset.Dataset
prepare
-
Constructor Details
-
AmazonReview
Creates a new instance ofAmazonReview
with the given necessary configurations.- Parameters:
builder
- a builder with the necessary configurations
-
-
Method Details
-
prepare
- Specified by:
prepare
in interfaceai.djl.training.dataset.Dataset
- Overrides:
prepare
in classCsvDataset
- Throws:
IOException
-
builder
Creates a new builder to build aAmazonReview
.- Returns:
- a new builder
-