Class MovieLens100k

  • All Implemented Interfaces:
    ai.djl.training.dataset.Dataset

    public final class MovieLens100k
    extends CsvDataset
    Movielens 100k movie reviews dataset from https://grouplens.org/datasets/movielens/100k/.
    • Method Detail

      • getCell

        public java.lang.String getCell​(long rowIndex,
                                        java.lang.String featureName)
        Returns a cell in the dataset.
        Overrides:
        getCell in class CsvDataset
        Parameters:
        rowIndex - the row index or record index for the cell
        featureName - the feature or column of the cell
        Returns:
        the value of the cell at that row and column
      • 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