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
public class AmazonReview extends CsvDataset
TheAmazonReview
dataset contains aApplication.NLP.SENTIMENT_ANALYSIS
set of reviews and their sentiment ratings.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
AmazonReview.Builder
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>>
-
-
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
-
-
Constructor Summary
Constructors Modifier Constructor Description protected
AmazonReview(AmazonReview.Builder builder)
Creates a new instance ofAmazonReview
with the given necessary configurations.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static 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, getRowFeatures, prepareFeaturizers
-
Methods inherited from class ai.djl.training.dataset.RandomAccessDataset
getData, getData, getData, getData, newSubDataset, newSubDataset, randomSplit, size, subDataset, subDataset, subDataset, subDataset, toArray
-
-
-
-
Constructor Detail
-
AmazonReview
protected AmazonReview(AmazonReview.Builder builder)
Creates a new instance ofAmazonReview
with the given necessary configurations.- Parameters:
builder
- a builder with the necessary configurations
-
-
Method Detail
-
prepare
public void prepare(ai.djl.util.Progress progress) throws java.io.IOException
- Specified by:
prepare
in interfaceai.djl.training.dataset.Dataset
- Overrides:
prepare
in classCsvDataset
- Throws:
java.io.IOException
-
builder
public static AmazonReview.Builder builder()
Creates a new builder to build aAmazonReview
.- Returns:
- a new builder
-
-