Package ai.djl.basicdataset.tabular
Class MovieLens100k
- java.lang.Object
-
- ai.djl.training.dataset.RandomAccessDataset
-
- ai.djl.basicdataset.tabular.TabularDataset
-
- ai.djl.basicdataset.tabular.CsvDataset
-
- ai.djl.basicdataset.tabular.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/.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classMovieLens100k.BuilderA builder to construct aMovieLens100k.-
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>>
-
-
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
-
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static MovieLens100k.Builderbuilder()Creates a builder to build aMovieLens100k.java.lang.StringgetCell(long rowIndex, java.lang.String featureName)Returns a cell in the dataset.voidprepare(ai.djl.util.Progress progress)-
Methods inherited from class ai.djl.basicdataset.tabular.CsvDataset
availableSize, 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
-
-
-
-
Method Detail
-
getCell
public java.lang.String getCell(long rowIndex, java.lang.String featureName)Returns a cell in the dataset.- Overrides:
getCellin classCsvDataset- Parameters:
rowIndex- the row index or record index for the cellfeatureName- 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:
preparein interfaceai.djl.training.dataset.Dataset- Overrides:
preparein classCsvDataset- Throws:
java.io.IOException
-
builder
public static MovieLens100k.Builder builder()
Creates a builder to build aMovieLens100k.- Returns:
- a new builder
-
-