Package ai.djl.basicdataset.tabular
Class MovieLens100k.Builder
java.lang.Object
ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<T>
ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder<T>
ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder<MovieLens100k.Builder>
ai.djl.basicdataset.tabular.MovieLens100k.Builder
- Enclosing class:
- MovieLens100k
public static final class MovieLens100k.Builder
extends CsvDataset.CsvBuilder<MovieLens100k.Builder>
A builder to construct a
MovieLens100k
.-
Field Summary
Fields inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
csvFormat, csvUrl
Fields inherited from class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
allowNoLabels, features, labels
Fields inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline
-
Method Summary
Modifier and TypeMethodDescriptionaddFeature
(String name) Adds a feature to the features set.build()
Builds the newMovieLens100k
.Returns the available features of this dataset.optArtifactId
(String artifactId) Sets the optional artifactId.optGroupId
(String groupId) Sets optional groupId.optRepository
(ai.djl.repository.Repository repository) Sets the optional repository.optUsage
(ai.djl.training.dataset.Dataset.Usage usage) Sets the optional usage.self()
Methods inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
optCsvFile, optCsvUrl, setCsvFormat
Methods inherited from class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
addCategoricalFeature, addCategoricalFeature, addCategoricalFeature, addCategoricalLabel, addCategoricalLabel, addCategoricalLabel, addFeature, addLabel, addNumericFeature, addNumericFeature, addNumericLabel, addNumericLabel, noLabels
Methods inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder
addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling
-
Method Details
-
self
- Overrides:
self
in classCsvDataset.CsvBuilder<MovieLens100k.Builder>
-
optUsage
Sets the optional usage.- Parameters:
usage
- the new usage- Returns:
- this builder
-
optRepository
Sets the optional repository.- Parameters:
repository
- the repository- Returns:
- this builder
-
optGroupId
Sets optional groupId.- Parameters:
groupId
- the groupId}- Returns:
- this builder
-
optArtifactId
Sets the optional artifactId.- Parameters:
artifactId
- the artifactId- Returns:
- this builder
-
getAvailableFeatures
Returns the available features of this dataset.- Returns:
- a list of feature names
-
addFeature
Adds a feature to the features set.- Parameters:
name
- the name of the feature- Returns:
- this builder
-
build
Builds the newMovieLens100k
.- Overrides:
build
in classCsvDataset.CsvBuilder<MovieLens100k.Builder>
- Returns:
- the new
MovieLens100k
-