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 final class	MovieLens100k.Builder	A builder to construct a MovieLens100k.

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

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.tabular.CsvDataset

csvFormat, csvRecords, csvUrl

Fields inherited from class ai.djl.basicdataset.tabular.TabularDataset

features, labels

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

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

Method Summary

Modifier and Type	Method	Description
static MovieLens100k.Builder	builder()	Creates a builder to build a MovieLens100k.
String	getCell(long rowIndex, String featureName)	Returns a cell in the dataset.
void	prepare(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, getRowDirect, 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

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

Method Details

getCell

public String getCell(long rowIndex,
 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 IOException

Specified by:

prepare in interface ai.djl.training.dataset.Dataset

Overrides:

prepare in class CsvDataset

Throws:

IOException

builder

public static MovieLens100k.Builder builder()

Creates a builder to build a MovieLens100k.

Returns:

a new builder