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