Class 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.training.dataset.RandomAccessDataset

        ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T extends ai.djl.training.dataset.RandomAccessDataset.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.training.dataset.RandomAccessDataset

        dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline
    • 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 a AmazonReview.
      void prepare​(ai.djl.util.Progress progress)
      • Methods inherited from class ai.djl.training.dataset.RandomAccessDataset

        getData, getData, getData, getData, randomSplit, size, 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