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

The AmazonReview dataset contains a Application.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 a AmazonReview.

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

Constructors

Modifier	Constructor	Description
protected	AmazonReview(AmazonReview.Builder builder)	Creates a new instance of AmazonReview with the given necessary configurations.

Method Summary

Modifier and Type	Method	Description
static AmazonReview.Builder	builder()	Creates a new builder to build a AmazonReview.
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, 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 Detail

AmazonReview

protected AmazonReview(AmazonReview.Builder builder)

Creates a new instance of AmazonReview 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 interface ai.djl.training.dataset.Dataset

Overrides:

prepare in class CsvDataset

Throws:

java.io.IOException

builder

public static AmazonReview.Builder builder()

Creates a new builder to build a AmazonReview.

Returns:

a new builder