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 aMovieLens100k
.
-
-
Field Summary
-
Fields inherited from class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
csvFormat, csvUrl
-
Fields inherited from class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
features, labels
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description MovieLens100k.Builder
addFeature(java.lang.String name)
Adds a feature to the features set.MovieLens100k
build()
Builds the newMovieLens100k
.java.util.List<java.lang.String>
getAvailableFeatures()
Returns the available features of this dataset.MovieLens100k.Builder
optArtifactId(java.lang.String artifactId)
Sets the optional artifactId.MovieLens100k.Builder
optGroupId(java.lang.String groupId)
Sets optional groupId.MovieLens100k.Builder
optRepository(ai.djl.repository.Repository repository)
Sets the optional repository.MovieLens100k.Builder
optUsage(ai.djl.training.dataset.Dataset.Usage usage)
Sets the optional usage.MovieLens100k.Builder
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, validate
-
-
-
-
Method Detail
-
self
public MovieLens100k.Builder self()
- Overrides:
self
in classCsvDataset.CsvBuilder<MovieLens100k.Builder>
-
optUsage
public MovieLens100k.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)
Sets the optional usage.- Parameters:
usage
- the new usage- Returns:
- this builder
-
optRepository
public MovieLens100k.Builder optRepository(ai.djl.repository.Repository repository)
Sets the optional repository.- Parameters:
repository
- the repository- Returns:
- this builder
-
optGroupId
public MovieLens100k.Builder optGroupId(java.lang.String groupId)
Sets optional groupId.- Parameters:
groupId
- the groupId}- Returns:
- this builder
-
optArtifactId
public MovieLens100k.Builder optArtifactId(java.lang.String artifactId)
Sets the optional artifactId.- Parameters:
artifactId
- the artifactId- Returns:
- this builder
-
getAvailableFeatures
public java.util.List<java.lang.String> getAvailableFeatures()
Returns the available features of this dataset.- Returns:
- a list of feature names
-
addFeature
public MovieLens100k.Builder addFeature(java.lang.String name)
Adds a feature to the features set.- Parameters:
name
- the name of the feature- Returns:
- this builder
-
build
public MovieLens100k build()
Builds the newMovieLens100k
.- Overrides:
build
in classCsvDataset.CsvBuilder<MovieLens100k.Builder>
- Returns:
- the new
MovieLens100k
-
-