ai.djl.basicdataset

Class StanfordMovieReview

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset

ai.djl.basicdataset.TextDataset

ai.djl.basicdataset.StanfordMovieReview

All Implemented Interfaces:

ai.djl.training.dataset.Dataset

public class StanfordMovieReview
extends TextDataset

The StanfordMovieReview dataset contains a Application.NLP.SENTIMENT_ANALYSIS set of movie reviews and their sentiment ratings.

The data is sourced from reviews located on IMDB (see here for details).

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	StanfordMovieReview.Builder A builder for a StanfordMovieReview.

Nested classes/interfaces inherited from class ai.djl.training.dataset.RandomAccessDataset

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

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.TextDataset

manager, prepared, resource, sourceTextData, targetTextData, usage

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

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

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
protected long	availableSize()
static StanfordMovieReview.Builder	builder() Creates a new builder to build a StanfordMovieReview.
protected ai.djl.training.dataset.Record	get(ai.djl.ndarray.NDManager manager, long index)
void	prepare(ai.djl.util.Progress progress)

Methods inherited from class ai.djl.basicdataset.TextDataset

getProcessedText, getRawText, getTextEmbedding, getVocabulary, preprocess

Methods inherited from class ai.djl.training.dataset.RandomAccessDataset

getData, getData, randomSplit, size

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

StanfordMovieReview

protected StanfordMovieReview(StanfordMovieReview.Builder builder)

Creates a new instance of StanfordMovieReview with the given necessary configurations.

Parameters:

builder - a builder with the necessary configurations

Method Detail

builder

public static StanfordMovieReview.Builder builder()

Creates a new builder to build a StanfordMovieReview.

Returns:

a new builder

prepare

public void prepare(ai.djl.util.Progress progress)
             throws java.io.IOException,
                    ai.djl.modality.nlp.embedding.EmbeddingException

Throws:

java.io.IOException

ai.djl.modality.nlp.embedding.EmbeddingException

get

protected ai.djl.training.dataset.Record get(ai.djl.ndarray.NDManager manager,
                                             long index)

Specified by:

get in class ai.djl.training.dataset.RandomAccessDataset

availableSize

protected long availableSize()

Specified by:

availableSize in class ai.djl.training.dataset.RandomAccessDataset