A B C D F G I L M N O P R S T U W 
All Classes All Packages

A

AbstractImageFolder - Class in ai.djl.basicdataset.cv.classification
A dataset for loading image files stored in a folder structure.
AbstractImageFolder(AbstractImageFolder.ImageFolderBuilder<?>) - Constructor for class ai.djl.basicdataset.cv.classification.AbstractImageFolder
 
AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder<T>> - Class in ai.djl.basicdataset.cv.classification
Used to build an AbstractImageFolder.
addAllFeatures() - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Adds all features to the features set.
addCategoricalFeature(String) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a categorical feature to the feature set.
addCategoricalFeature(String, boolean) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a categorical feature to the feature set.
addCategoricalFeature(String, Map<String, Integer>, boolean) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a categorical feature to the feature set with specified mapping.
addCategoricalLabel(String) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a categorical feature to the label set.
addCategoricalLabel(String, boolean) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a categorical feature to the label set.
addCategoricalLabel(String, Map<String, Integer>, boolean) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a categorical feature to the feature set with specified mapping.
addFeature(Feature...) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds the features to the feature set.
addFeature(String) - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Adds a feature to the features set.
addFeature(String) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Adds a feature to the features set.
addFeature(String) - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Adds a feature to the features set.
addFeature(String) - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
Adds a feature to the features set.
addFeature(String, boolean) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Adds a feature to the features set with onehot encoding.
addLabel(Feature...) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds the features to the label set.
addNumericFeature(String) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a numeric feature to the feature set.
addNumericFeature(String, boolean) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a numeric feature to the feature set.
addNumericLabel(String) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a number feature to the label set.
addNumericLabel(String, boolean) - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Adds a number feature to the label set.
ai.djl.basicdataset - package ai.djl.basicdataset
Contains a library of built-in datasets.
ai.djl.basicdataset.cv - package ai.djl.basicdataset.cv
Contains a library of built-in datasets for Application.CV.
ai.djl.basicdataset.cv.classification - package ai.djl.basicdataset.cv.classification
Contains a library of built-in datasets for Application.CV.IMAGE_CLASSIFICATION.
ai.djl.basicdataset.nlp - package ai.djl.basicdataset.nlp
Contains a library of built-in datasets for Application.NLP.
ai.djl.basicdataset.tabular - package ai.djl.basicdataset.tabular
Contains a library of built-in datasets for Application.Tabular.
ai.djl.basicdataset.tabular.utils - package ai.djl.basicdataset.tabular.utils
Contains utilities used within datasets that are Application.Tabular.
ai.djl.basicdataset.utils - package ai.djl.basicdataset.utils
Contains utilities used within the basic datasets.
AirfoilRandomAccess - Class in ai.djl.basicdataset.tabular
AirfoilRandomAccess.Builder - Class in ai.djl.basicdataset.tabular
A builder to construct a AirfoilRandomAccess.
allowNoLabels - Variable in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
 
AmazonReview - Class in ai.djl.basicdataset.nlp
The AmazonReview dataset contains a Application.NLP.SENTIMENT_ANALYSIS set of reviews and their sentiment ratings.
AmazonReview(AmazonReview.Builder) - Constructor for class ai.djl.basicdataset.nlp.AmazonReview
Creates a new instance of AmazonReview with the given necessary configurations.
AmazonReview.Builder - Class in ai.djl.basicdataset.nlp
A builder to construct a AmazonReview.
AmesRandomAccess - Class in ai.djl.basicdataset.tabular
Ames house pricing dataset from https://www.kaggle.com/c/house-prices-advanced-regression-techniques/data.
AmesRandomAccess.Builder - Class in ai.djl.basicdataset.tabular
A builder to construct a AmesRandomAccess.
Annotation() - Constructor for class ai.djl.basicdataset.cv.CocoMetadata.Annotation
 
apply(T) - Method in interface ai.djl.basicdataset.utils.ThrowingFunction
Applies this function to the given argument.
artifactId - Variable in class ai.djl.basicdataset.nlp.TextDataset.Builder
 
availableSize() - Method in class ai.djl.basicdataset.cv.BananaDetection
availableSize() - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
availableSize() - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset
availableSize() - Method in class ai.djl.basicdataset.cv.CocoDetection
availableSize() - Method in class ai.djl.basicdataset.cv.PikachuDetection
availableSize() - Method in class ai.djl.basicdataset.nlp.GoEmotions
Returns the number of records available to be read in this Dataset.
availableSize() - Method in class ai.djl.basicdataset.nlp.PennTreebankText
availableSize() - Method in class ai.djl.basicdataset.nlp.StanfordMovieReview
availableSize() - Method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset
Returns the number of records available to be read in this Dataset.
availableSize() - Method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset
availableSize() - Method in class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT
Returns the number of records available to be read in this Dataset.
availableSize() - Method in class ai.djl.basicdataset.tabular.CsvDataset

B

BananaDetection - Class in ai.djl.basicdataset.cv
Banana image detection dataset contains a 3 x 256 x 256 image in the dataset with a banana of varying sizes in each image.
BananaDetection(BananaDetection.Builder) - Constructor for class ai.djl.basicdataset.cv.BananaDetection
Creates a new instance of RandomAccessDataset with the given necessary configurations.
BananaDetection.Builder - Class in ai.djl.basicdataset.cv
A builder for a BananaDetection.
BaseBuilder() - Constructor for class ai.djl.basicdataset.cv.ImageDataset.BaseBuilder
 
BaseBuilder() - Constructor for class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
 
BasicDatasets - Interface in ai.djl.basicdataset
An interface which contains datasets that are hosted on https://mlrepo.djl.ai/.
build() - Method in class ai.djl.basicdataset.cv.BananaDetection.Builder
Builds the BananaDetection.
build() - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset.Builder
Builds the CaptchaDataset.
build() - Method in class ai.djl.basicdataset.cv.classification.Cifar10.Builder
Builds a new Cifar10.
build() - Method in class ai.djl.basicdataset.cv.classification.FashionMnist.Builder
Builds the Mnist.
build() - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten.Builder
build() - Method in class ai.djl.basicdataset.cv.classification.ImageFolder.Builder
Builds the ImageFolder.
build() - Method in class ai.djl.basicdataset.cv.classification.ImageNet.Builder
Builds the ImageNet.
build() - Method in class ai.djl.basicdataset.cv.classification.Mnist.Builder
Builds the Mnist.
build() - Method in class ai.djl.basicdataset.cv.CocoDetection.Builder
Builds the new CocoDetection.
build() - Method in class ai.djl.basicdataset.cv.PikachuDetection.Builder
Builds the PikachuDetection.
build() - Method in class ai.djl.basicdataset.nlp.AmazonReview.Builder
Builds the new CsvDataset.
build() - Method in class ai.djl.basicdataset.nlp.CookingStackExchange.Builder
Builds a new CookingStackExchange.
build() - Method in class ai.djl.basicdataset.nlp.GoEmotions.Builder
build() - Method in class ai.djl.basicdataset.nlp.PennTreebankText.Builder
Builds a new PennTreebankText object.
build() - Method in class ai.djl.basicdataset.nlp.StanfordMovieReview.Builder
build() - Method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset.Builder
build() - Method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset.Builder
build() - Method in class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT.Builder
build() - Method in class ai.djl.basicdataset.nlp.WikiText2.Builder
Builds a new WikiText2 object.
build() - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Builds the new CsvDataset.
build() - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Builds the new AmesRandomAccess.
build() - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
Builds the new CsvDataset.
build() - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Builds the new DailyDelhiClimate.
build() - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
Builds the new MovieLens100k.
buildBaseTranslator(Model, Map<String, ?>) - Method in class ai.djl.basicdataset.tabular.TabularTranslatorFactory
builder() - Static method in class ai.djl.basicdataset.cv.BananaDetection
Creates a new builder to build a BananaDetection.
builder() - Static method in class ai.djl.basicdataset.cv.classification.CaptchaDataset
Creates a builder to build a CaptchaDataset.
builder() - Static method in class ai.djl.basicdataset.cv.classification.Cifar10
Creates a builder to build a Cifar10.
builder() - Static method in class ai.djl.basicdataset.cv.classification.FashionMnist
Creates a builder to build a Mnist.
builder() - Static method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten
Creates a new builder to build a FruitsFreshAndRotten.
builder() - Static method in class ai.djl.basicdataset.cv.classification.ImageFolder
Creates a new builder to build a ImageFolder.
builder() - Static method in class ai.djl.basicdataset.cv.classification.ImageNet
Creates a new builder to build a ImageNet.
builder() - Static method in class ai.djl.basicdataset.cv.classification.Mnist
Creates a builder to build a Mnist.
builder() - Static method in class ai.djl.basicdataset.cv.CocoDetection
Creates a builder to build a CocoDetection.
builder() - Static method in class ai.djl.basicdataset.cv.PikachuDetection
Creates a new builder to build a PikachuDetection.
builder() - Static method in class ai.djl.basicdataset.nlp.AmazonReview
Creates a new builder to build a AmazonReview.
builder() - Static method in class ai.djl.basicdataset.nlp.CookingStackExchange
Creates a builder to build a CookingStackExchange.
builder() - Static method in class ai.djl.basicdataset.nlp.GoEmotions
Creates a builder to build a GoEmotions.
builder() - Static method in class ai.djl.basicdataset.nlp.PennTreebankText
Creates a builder to build a PennTreebankText.
builder() - Static method in class ai.djl.basicdataset.nlp.StanfordMovieReview
Creates a new builder to build a StanfordMovieReview.
builder() - Static method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset
Creates a new builder to build a StanfordQuestionAnsweringDataset.
builder() - Static method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset
Creates a new builder to build a TatoebaEnglishFrenchDataset.
builder() - Static method in class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT
Creates a new builder to build a UniversalDependenciesEnglishEWT.
builder() - Static method in class ai.djl.basicdataset.nlp.WikiText2
Creates a builder to build a WikiText2.
builder() - Static method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess
Creates a builder to build a AirfoilRandomAccess.
builder() - Static method in class ai.djl.basicdataset.tabular.AmesRandomAccess
Creates a builder to build a AmesRandomAccess.
builder() - Static method in class ai.djl.basicdataset.tabular.CsvDataset
Creates a builder to build a AmesRandomAccess.
builder() - Static method in class ai.djl.basicdataset.tabular.DailyDelhiClimate
Creates a builder to build a DailyDelhiClimate.
builder() - Static method in class ai.djl.basicdataset.tabular.MovieLens100k
Creates a builder to build a MovieLens100k.
Builder() - Constructor for class ai.djl.basicdataset.nlp.GoEmotions.Builder
Constructs a new builder.
Builder() - Constructor for class ai.djl.basicdataset.nlp.PennTreebankText.Builder
Constructs a new builder.
Builder() - Constructor for class ai.djl.basicdataset.nlp.StanfordMovieReview.Builder
Constructs a new builder.
Builder() - Constructor for class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset.Builder
Constructs a new builder.
Builder() - Constructor for class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset.Builder
Constructs a new builder.
Builder() - Constructor for class ai.djl.basicdataset.nlp.TextDataset.Builder
Constructs a new builder.
Builder() - Constructor for class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT.Builder
Constructs a new builder.

C

CAPTCHA_LENGTH - Static variable in class ai.djl.basicdataset.cv.classification.CaptchaDataset
 
CAPTCHA_OPTIONS - Static variable in class ai.djl.basicdataset.cv.classification.CaptchaDataset
 
CaptchaDataset - Class in ai.djl.basicdataset.cv.classification
A Dataset featuring captcha images.
CaptchaDataset(CaptchaDataset.Builder) - Constructor for class ai.djl.basicdataset.cv.classification.CaptchaDataset
Creates a new instance of CaptchaDataset.
CaptchaDataset.Builder - Class in ai.djl.basicdataset.cv.classification
A builder for a CaptchaDataset.
Category() - Constructor for class ai.djl.basicdataset.cv.CocoMetadata.Category
 
Cifar10 - Class in ai.djl.basicdataset.cv.classification
CIFAR10 image classification dataset from https://www.cs.toronto.edu/~kriz/cifar.html.
Cifar10.Builder - Class in ai.djl.basicdataset.cv.classification
A builder to construct a Cifar10.
CocoDetection - Class in ai.djl.basicdataset.cv
Coco image detection dataset from http://cocodataset.org/#home.
CocoDetection.Builder - Class in ai.djl.basicdataset.cv
A builder to construct a CocoDetection.
CocoMetadata - Class in ai.djl.basicdataset.cv
A metadata class to represent the structure of annotations in Coco.
CocoMetadata() - Constructor for class ai.djl.basicdataset.cv.CocoMetadata
 
CocoMetadata.Annotation - Class in ai.djl.basicdataset.cv
An annotation applied to an image in the coco dataset.
CocoMetadata.Category - Class in ai.djl.basicdataset.cv
An annotation category in the coco dataset.
CocoMetadata.Image - Class in ai.djl.basicdataset.cv
An image in the coco dataset.
CocoUtils - Class in ai.djl.basicdataset.cv
A utility class that assists in loading and parsing the annotations in Coco.
Configuration() - Constructor for class ai.djl.basicdataset.utils.TextData.Configuration
 
CookingStackExchange - Class in ai.djl.basicdataset.nlp
A text classification dataset contains questions from cooking.stackexchange.com and their associated tags on the site.
CookingStackExchange.Builder - Class in ai.djl.basicdataset.nlp
A builder to construct a CookingStackExchange.
CsvBuilder() - Constructor for class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
 
CsvDataset - Class in ai.djl.basicdataset.tabular
CsvDataset represents the dataset that stored in a .csv file.
CsvDataset(CsvDataset.CsvBuilder<?>) - Constructor for class ai.djl.basicdataset.tabular.CsvDataset
 
CsvDataset.CsvBuilder<T extends CsvDataset.CsvBuilder<T>> - Class in ai.djl.basicdataset.tabular
Used to build a CsvDataset.
csvFormat - Variable in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
 
csvFormat - Variable in class ai.djl.basicdataset.tabular.CsvDataset
 
csvRecords - Variable in class ai.djl.basicdataset.tabular.CsvDataset
 
csvUrl - Variable in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
 
csvUrl - Variable in class ai.djl.basicdataset.tabular.CsvDataset
 

D

DailyDelhiClimate - Class in ai.djl.basicdataset.tabular
DailyDelhiClimate.Builder - Class in ai.djl.basicdataset.tabular
A builder to construct a DailyDelhiClimate.
dataRequired() - Method in interface ai.djl.basicdataset.tabular.utils.Featurizer.DataFeaturizer
Returns the length of the data array required by Featurizer.deFeaturize(float[]).
dataRequired() - Method in interface ai.djl.basicdataset.tabular.utils.Featurizer
Returns the length of the data array required by Featurizer.deFeaturize(float[]).
deFeaturize(float[]) - Method in interface ai.djl.basicdataset.tabular.utils.Featurizer.DataFeaturizer
Converts the output data for a label back into the Java type.
deFeaturize(float[]) - Method in interface ai.djl.basicdataset.tabular.utils.Featurizer
Converts the output data for a label back into the Java type.
DJL_REPO_URL - Static variable in interface ai.djl.basicdataset.BasicDatasets
 
DynamicBuffer - Class in ai.djl.basicdataset.tabular.utils
A float buffer that can dynamically change it's capacity.
DynamicBuffer() - Constructor for class ai.djl.basicdataset.tabular.utils.DynamicBuffer
Constructs a new instance of DynamicBuffer.

F

FashionMnist - Class in ai.djl.basicdataset.cv.classification
FashMnist is a dataset from Zalando article images (https://github.com/zalandoresearch/fashion-mnist).
FashionMnist.Builder - Class in ai.djl.basicdataset.cv.classification
A builder for a FashionMnist.
Feature - Class in ai.djl.basicdataset.tabular.utils
A class contains feature name and its Featurizer.
Feature(String, boolean) - Constructor for class ai.djl.basicdataset.tabular.utils.Feature
Constructs a Feature instance.
Feature(String, Featurizer) - Constructor for class ai.djl.basicdataset.tabular.utils.Feature
Constructs a Feature instance.
Feature(String, Featurizer.DataFeaturizer) - Constructor for class ai.djl.basicdataset.tabular.utils.Feature
Constructs a Feature instance.
Feature(String, Map<String, Integer>, boolean) - Constructor for class ai.djl.basicdataset.tabular.utils.Feature
Constructs a Feature instance.
features - Variable in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
 
features - Variable in class ai.djl.basicdataset.tabular.TabularDataset
 
featurize(DynamicBuffer, String) - Method in interface ai.djl.basicdataset.tabular.utils.Featurizer
Puts encoded data into the float buffer.
Featurizer - Interface in ai.djl.basicdataset.tabular.utils
An interface that convert String to numeric data.
Featurizer.DataFeaturizer - Interface in ai.djl.basicdataset.tabular.utils
A Featurizer that only supports the data featurize operations, but not the full deFeaturize operations used by labels.
Featurizers - Class in ai.djl.basicdataset.tabular.utils
A utility class provides helper functions to create Featurizer.
FixedBucketSampler - Class in ai.djl.basicdataset.utils
FixedBucketSampler is a Sampler to be used with TextDataset, and PaddingStackBatchifier.
FixedBucketSampler(int) - Constructor for class ai.djl.basicdataset.utils.FixedBucketSampler
Constructs a new instance of FixedBucketSampler with the given number of buckets, and the given batch size.
FixedBucketSampler(int, int) - Constructor for class ai.djl.basicdataset.utils.FixedBucketSampler
Constructs a new instance of FixedBucketSampler with the given number of buckets, and the given batch size.
FixedBucketSampler(int, int, boolean) - Constructor for class ai.djl.basicdataset.utils.FixedBucketSampler
Constructs a new instance of FixedBucketSampler with the given number of buckets, and the given batch size.
flag - Variable in class ai.djl.basicdataset.cv.ImageDataset
 
fromMap(Map<String, String>) - Static method in class ai.djl.basicdataset.tabular.MapFeatures
Creates a MapFeatures from a source list.
FruitsFreshAndRotten - Class in ai.djl.basicdataset.cv.classification
FruitRottenFresh classification dataset that contains the same fruit where rotten and fresh class are stored in different sub folders.
FruitsFreshAndRotten.Builder - Class in ai.djl.basicdataset.cv.classification
A builder for the FruitsFreshAndRotten.

G

get(NDManager, long) - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset
get(NDManager, long) - Method in class ai.djl.basicdataset.cv.classification.ImageClassificationDataset
get(NDManager, long) - Method in class ai.djl.basicdataset.cv.ObjectDetectionDataset
get(NDManager, long) - Method in class ai.djl.basicdataset.nlp.GoEmotions
Gets the Record for the given index from the dataset.
get(NDManager, long) - Method in class ai.djl.basicdataset.nlp.PennTreebankText
get(NDManager, long) - Method in class ai.djl.basicdataset.nlp.StanfordMovieReview
get(NDManager, long) - Method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset
Gets the Record for the given index from the dataset.
get(NDManager, long) - Method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset
get(NDManager, long) - Method in class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT
Gets the Record for the given index from the dataset.
get(NDManager, long) - Method in class ai.djl.basicdataset.tabular.TabularDataset
getAll() - Method in class ai.djl.basicdataset.tabular.TabularResults
Returns all of the TabularResults.TabularResult.
getAnnotationById(long) - Method in class ai.djl.basicdataset.cv.CocoUtils
Returns an CocoMetadata.Annotation that corresponds to a given annotation id.
getAnnotationIdByImageId(long) - Method in class ai.djl.basicdataset.cv.CocoUtils
Returns all ids of the annotation that correspond to a given image id.
getAnnotations() - Method in class ai.djl.basicdataset.cv.CocoMetadata
Returns a list of all annotations.
getArea() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Annotation
Returns the area of this annotation.
getAvailableFeatures() - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Returns the available features of this dataset.
getAvailableFeatures() - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Returns the available features of this dataset.
getAvailableFeatures() - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Returns the available features of this dataset.
getAvailableFeatures() - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
Returns the available features of this dataset.
getBaseInputType() - Method in class ai.djl.basicdataset.tabular.TabularTranslatorFactory
getBaseOutputType() - Method in class ai.djl.basicdataset.tabular.TabularTranslatorFactory
getBatchSize() - Method in class ai.djl.basicdataset.utils.FixedBucketSampler
getBoundingBox() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Annotation
Returns the bounding box of this annotation.
getBuffer() - Method in class ai.djl.basicdataset.tabular.utils.DynamicBuffer
Returns a FloatBuffer that contains all the data.
getCategories() - Method in class ai.djl.basicdataset.cv.CocoMetadata
Returns a list of all categories.
getCategoryId() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Annotation
Returns the category id of this annotation.
getCell(long, String) - Method in class ai.djl.basicdataset.tabular.CsvDataset
Returns a cell in the dataset.
getCell(long, String) - Method in class ai.djl.basicdataset.tabular.MovieLens100k
Returns a cell in the dataset.
getCell(long, String) - Method in class ai.djl.basicdataset.tabular.TabularDataset
Returns a cell in the dataset.
getClasses() - Method in class ai.djl.basicdataset.cv.BananaDetection
Returns the classes that detected objects in the dataset can be classified into.
getClasses() - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
Returns the classes that the images in the dataset are classified into.
getClasses() - Method in class ai.djl.basicdataset.cv.classification.ImageClassificationDataset
Returns the classes that the images in the dataset are classified into.
getClasses() - Method in class ai.djl.basicdataset.cv.CocoDetection
Returns the classes that detected objects in the dataset can be classified into.
getClasses() - Method in class ai.djl.basicdataset.cv.ObjectDetectionDataset
Returns the classes that detected objects in the dataset can be classified into.
getClasses() - Method in class ai.djl.basicdataset.cv.PikachuDetection
Returns the classes that detected objects in the dataset can be classified into.
getClassFull() - Method in class ai.djl.basicdataset.cv.classification.ImageNet
Returns all full class names of this ImageNet dataset.
getClassNames() - Method in class ai.djl.basicdataset.cv.classification.ImageNet
Returns all class names of this ImageNet dataset.
getClassNumber(long) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
Returns the class of the data item at the given index.
getClassNumber(long) - Method in class ai.djl.basicdataset.cv.classification.ImageClassificationDataset
Returns the class of the data item at the given index.
getCocoUrl() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Image
Returns the url of this image.
getColumnNames() - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess
Returns the column names of the CSV file.
getColumnNames() - Method in class ai.djl.basicdataset.tabular.CsvDataset
Returns the column names of the CSV file.
getData() - Method in class ai.djl.basicdataset.nlp.CookingStackExchange
getData() - Method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset
Get data from the SQuAD dataset.
getData() - Method in class ai.djl.basicdataset.nlp.WikiText2
Get data from the WikiText2 dataset.
getData(NDManager) - Method in class ai.djl.basicdataset.nlp.CookingStackExchange
getData(NDManager) - Method in class ai.djl.basicdataset.nlp.WikiText2
Fetches an iterator that can iterate through the Dataset.
getDefaultConfiguration() - Static method in class ai.djl.basicdataset.utils.TextData
Returns a good default TextData.Configuration to use for the constructor with defaults.
getEmbedding(NDManager, long) - Method in class ai.djl.basicdataset.utils.TextData
Gets the text embedding for the given index of the text input.
getEpochDayFeaturizer(String) - Static method in class ai.djl.basicdataset.tabular.utils.Featurizers
Constructs an Featurizers.EpochDayFeaturizer for representing dates using the epoch day (number of days since 1970-01-01).
getExpansions() - Method in class ai.djl.basicdataset.tabular.TabularTranslator
getFeature(int) - Method in class ai.djl.basicdataset.tabular.TabularResults
Returns the result for the given feature index.
getFeature(String) - Method in class ai.djl.basicdataset.tabular.TabularResults
Returns the result for the given feature name.
getFeatures() - Method in class ai.djl.basicdataset.tabular.TabularDataset
Returns the dataset features.
getFeatures() - Method in class ai.djl.basicdataset.tabular.TabularTranslator
Returns the features for the translator.
getFeatureSize() - Method in class ai.djl.basicdataset.tabular.TabularDataset
Gets the feature size of current TabularDataset.
getFeaturizer() - Method in class ai.djl.basicdataset.tabular.utils.Feature
Returns the Featurizer.
getHeight() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Image
Returns the height of this image.
getId() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Annotation
Returns the id of this annotation.
getId() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Category
Returns the id of this category.
getId() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Image
Returns the id of this image.
getImage(long) - Method in class ai.djl.basicdataset.cv.BananaDetection
Returns the image at the given index in the dataset.
getImage(long) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
Returns the image at the given index in the dataset.
getImage(long) - Method in class ai.djl.basicdataset.cv.CocoDetection
 
getImage(long) - Method in class ai.djl.basicdataset.cv.ImageDataset
Returns the image at the given index in the dataset.
getImage(long) - Method in class ai.djl.basicdataset.cv.PikachuDetection
 
getImageChannels() - Method in class ai.djl.basicdataset.cv.ImageDataset
Returns the number of channels in the images in the dataset.
getImageHeight() - Method in class ai.djl.basicdataset.cv.BananaDetection
Returns the height of the images in the dataset.
getImageHeight() - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
Returns the height of the images in the dataset.
getImageHeight() - Method in class ai.djl.basicdataset.cv.CocoDetection
Returns the height of the images in the dataset.
getImageHeight() - Method in class ai.djl.basicdataset.cv.ImageDataset
Returns the height of the images in the dataset.
getImageHeight() - Method in class ai.djl.basicdataset.cv.PikachuDetection
Returns the height of the images in the dataset.
getImageId() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Annotation
Returns the id of the image this annotation applies to.
getImageIds() - Method in class ai.djl.basicdataset.cv.CocoUtils
Returns all image ids in the annotation file.
getImagePath(String) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
 
getImagePath(String) - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten
getImagePath(String) - Method in class ai.djl.basicdataset.cv.classification.ImageFolder
getImagePath(String) - Method in class ai.djl.basicdataset.cv.classification.ImageNet
getImages() - Method in class ai.djl.basicdataset.cv.CocoMetadata
Returns a list of all images.
getImageWidth() - Method in class ai.djl.basicdataset.cv.BananaDetection
Returns the width of the images in the dataset.
getImageWidth() - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
Returns the width of the images in the dataset.
getImageWidth() - Method in class ai.djl.basicdataset.cv.CocoDetection
Returns the width of the images in the dataset.
getImageWidth() - Method in class ai.djl.basicdataset.cv.ImageDataset
Returns the width of the images in the dataset.
getImageWidth() - Method in class ai.djl.basicdataset.cv.PikachuDetection
Returns the width of the images in the dataset.
getIndex() - Method in class ai.djl.basicdataset.nlp.TextDataset.Sample
Returns the sample index.
getLabels() - Method in class ai.djl.basicdataset.tabular.TabularDataset
Returns the dataset labels.
getLabels() - Method in class ai.djl.basicdataset.tabular.TabularTranslator
Returns the labels for the translator.
getLabelSize() - Method in class ai.djl.basicdataset.tabular.TabularDataset
Gets the label size of current TabularDataset.
getLength() - Method in class ai.djl.basicdataset.tabular.utils.DynamicBuffer
Returns the buffer size.
getName() - Method in class ai.djl.basicdataset.tabular.TabularResults.TabularResult
Returns the result (feature) name.
getName() - Method in class ai.djl.basicdataset.tabular.utils.Feature
Returns the feature name.
getNumericFeaturizer() - Static method in class ai.djl.basicdataset.tabular.utils.Featurizers
Returns the default numeric Featurizer.
getNumericFeaturizer(boolean) - Static method in class ai.djl.basicdataset.tabular.utils.Featurizers
Returns the default numeric Featurizer.
getObjects(long) - Method in class ai.djl.basicdataset.cv.BananaDetection
Returns the list of objects in the image at the given index.
getObjects(long) - Method in class ai.djl.basicdataset.cv.CocoDetection
Returns the list of objects in the image at the given index.
getObjects(long) - Method in class ai.djl.basicdataset.cv.ObjectDetectionDataset
Returns the list of objects in the image at the given index.
getObjects(long) - Method in class ai.djl.basicdataset.cv.PikachuDetection
Returns the list of objects in the image at the given index.
getPostprocessorExpansions() - Method in class ai.djl.basicdataset.tabular.TabularTranslatorFactory
getPreprocessorExpansions() - Method in class ai.djl.basicdataset.tabular.TabularTranslatorFactory
getProcessedText(long) - Method in class ai.djl.basicdataset.utils.TextData
Gets the textual input after preprocessing.
getProcessedText(long, boolean) - Method in class ai.djl.basicdataset.nlp.TextDataset
Gets the processed textual input.
getRawText(long) - Method in class ai.djl.basicdataset.utils.TextData
Gets the raw textual input.
getRawText(long, boolean) - Method in class ai.djl.basicdataset.nlp.TextDataset
Gets the raw textual input.
getRecordImage(NDManager, long) - Method in class ai.djl.basicdataset.cv.ImageDataset
 
getRelativeImagePath(long) - Method in class ai.djl.basicdataset.cv.CocoUtils
Returns the relative path of an image given an image id.
getResult() - Method in class ai.djl.basicdataset.tabular.TabularResults.TabularResult
Returns the computed result.
getRowFeatures(NDManager, long, List<Feature>) - Method in class ai.djl.basicdataset.tabular.TabularDataset
Returns the designated features (either data or label features) from a row.
getSamples() - Method in class ai.djl.basicdataset.nlp.TextDataset
Returns a list of sample information.
getSentenceLength() - Method in class ai.djl.basicdataset.nlp.TextDataset.Sample
Returns the sentence length.
getSize() - Method in class ai.djl.basicdataset.utils.TextData
Returns the size of the data.
getStringFeaturizer() - Static method in class ai.djl.basicdataset.tabular.utils.Featurizers
Returns the default String Featurizer.
getStringFeaturizer(boolean) - Static method in class ai.djl.basicdataset.tabular.utils.Featurizers
Returns the default String Featurizer.
getStringFeaturizer(Map<String, Integer>, boolean) - Static method in class ai.djl.basicdataset.tabular.utils.Featurizers
Returns a new instance of String Featurizer.
getSynset() - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
Returns the synsets of the ImageFolder dataset.
getTextEmbedding() - Method in class ai.djl.basicdataset.utils.TextData
Gets the TextEmbedding used to embed the data with.
getTextEmbedding(boolean) - Method in class ai.djl.basicdataset.nlp.TextDataset
Gets the word embedding used while pre-processing the dataset.
getVocabulary() - Method in class ai.djl.basicdataset.utils.TextData
Gets the DefaultVocabulary built while preprocessing the text data.
getVocabulary(boolean) - Method in class ai.djl.basicdataset.nlp.TextDataset
Gets the DefaultVocabulary built while preprocessing the text data.
getWidth() - Method in class ai.djl.basicdataset.cv.CocoMetadata.Image
Returns the width of this image.
getWordNetIds() - Method in class ai.djl.basicdataset.cv.classification.ImageNet
Returns all WordNet ids of this ImageNet dataset.
GoEmotions - Class in ai.djl.basicdataset.nlp
GoEmotions is a corpus of 58k carefully curated comments extracted from Reddit, with human annotations to 27 emotion categories or Neutral.
GoEmotions.Builder - Class in ai.djl.basicdataset.nlp
A builder to construct a GoEmotions.
GROUP_ID - Static variable in interface ai.djl.basicdataset.BasicDatasets
 
groupId - Variable in class ai.djl.basicdataset.nlp.TextDataset.Builder
 

I

Image() - Constructor for class ai.djl.basicdataset.cv.CocoMetadata.Image
 
IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.cv.classification.CaptchaDataset
 
IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.cv.classification.Cifar10
 
IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.cv.classification.FashionMnist
 
IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.cv.classification.Mnist
 
IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.cv.classification.CaptchaDataset
 
IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.cv.classification.Cifar10
 
IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.cv.classification.FashionMnist
 
IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.cv.classification.Mnist
 
ImageClassificationDataset - Class in ai.djl.basicdataset.cv.classification
A helper to create Datasets for Application.CV.IMAGE_CLASSIFICATION.
ImageClassificationDataset(ImageDataset.BaseBuilder<?>) - Constructor for class ai.djl.basicdataset.cv.classification.ImageClassificationDataset
Creates a new instance of RandomAccessDataset with the given necessary configurations.
ImageDataset - Class in ai.djl.basicdataset.cv
A helper to create a Dataset where the data contains a single image.
ImageDataset(ImageDataset.BaseBuilder<?>) - Constructor for class ai.djl.basicdataset.cv.ImageDataset
Creates a new instance of RandomAccessDataset with the given necessary configurations.
ImageDataset.BaseBuilder<T extends ImageDataset.BaseBuilder<T>> - Class in ai.djl.basicdataset.cv
Used to build an ImageClassificationDataset.
ImageFolder - Class in ai.djl.basicdataset.cv.classification
A dataset for loading image files stored in a folder structure.
ImageFolder.Builder - Class in ai.djl.basicdataset.cv.classification
A builder for the ImageFolder.
ImageFolderBuilder() - Constructor for class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
 
ImageNet - Class in ai.djl.basicdataset.cv.classification
ImageNet is an image classification dataset from http://image-net.org 2012 Classification dataset.
ImageNet.Builder - Class in ai.djl.basicdataset.cv.classification
A builder for a ImageNet.
items - Variable in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
 

L

labels - Variable in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
 
labels - Variable in class ai.djl.basicdataset.tabular.TabularDataset
 
ListFeatures - Class in ai.djl.basicdataset.tabular
An extension of the ArrayList for use in the TabularTranslator.
ListFeatures() - Constructor for class ai.djl.basicdataset.tabular.ListFeatures
Constructs a ListFeatures instance.
ListFeatures(int) - Constructor for class ai.djl.basicdataset.tabular.ListFeatures
Constructs a ListFeatures instance.
ListFeatures(Collection<? extends String>) - Constructor for class ai.djl.basicdataset.tabular.ListFeatures
Constructs a ListFeatures instance.
ListFeatures(List<String>) - Constructor for class ai.djl.basicdataset.tabular.ListFeatures
Constructs a ListFeatures instance from a source list.
listImages(Path, List<String>) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
 

M

manager - Variable in class ai.djl.basicdataset.nlp.TextDataset
 
mapCategoryId(long) - Method in class ai.djl.basicdataset.cv.CocoUtils
Returns the continuous category id given an original category id.
MapFeatures - Class in ai.djl.basicdataset.tabular
An extension of the ConcurrentHashMap for use in the TabularTranslator.
MapFeatures() - Constructor for class ai.djl.basicdataset.tabular.MapFeatures
Constructs a MapFeatures instance.
MapFeatures(int) - Constructor for class ai.djl.basicdataset.tabular.MapFeatures
Constructs a MapFeatures instance.
MapFeatures(int, float) - Constructor for class ai.djl.basicdataset.tabular.MapFeatures
Constructs a MapFeatures instance.
MapFeatures(int, float, int) - Constructor for class ai.djl.basicdataset.tabular.MapFeatures
Constructs a MapFeatures.
MapFeatures(Map<? extends String, ? extends String>) - Constructor for class ai.djl.basicdataset.tabular.MapFeatures
Constructs a MapFeatures instance.
matchingTranslatorOptions() - Method in class ai.djl.basicdataset.cv.classification.ImageClassificationDataset
matchingTranslatorOptions() - Method in class ai.djl.basicdataset.tabular.TabularDataset
Mnist - Class in ai.djl.basicdataset.cv.classification
MNIST handwritten digits dataset from http://yann.lecun.com/exdb/mnist.
Mnist.Builder - Class in ai.djl.basicdataset.cv.classification
A builder for a Mnist.
MovieLens100k - Class in ai.djl.basicdataset.tabular
Movielens 100k movie reviews dataset from https://grouplens.org/datasets/movielens/100k/.
MovieLens100k.Builder - Class in ai.djl.basicdataset.tabular
A builder to construct a MovieLens100k.
mrl - Variable in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
 
mrl - Variable in class ai.djl.basicdataset.nlp.TextDataset
 

N

noLabels() - Method in class ai.djl.basicdataset.tabular.TabularDataset.BaseBuilder
Indicates the dataset should not have any labels.
NORMALIZE_MEAN - Static variable in class ai.djl.basicdataset.cv.classification.Cifar10
 
NORMALIZE_STD - Static variable in class ai.djl.basicdataset.cv.classification.Cifar10
 
NUM_CLASSES - Static variable in class ai.djl.basicdataset.cv.classification.FashionMnist
 
NUM_CLASSES - Static variable in class ai.djl.basicdataset.cv.classification.Mnist
 

O

ObjectDetectionDataset - Class in ai.djl.basicdataset.cv
A helper to create Datasets for Application.CV.OBJECT_DETECTION.
ObjectDetectionDataset(ImageDataset.BaseBuilder<?>) - Constructor for class ai.djl.basicdataset.cv.ObjectDetectionDataset
Creates a new instance of ObjectDetectionDataset with the given necessary configurations.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.BananaDetection.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.classification.Cifar10.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.classification.FashionMnist.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.classification.Mnist.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.CocoDetection.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.cv.PikachuDetection.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.nlp.AmazonReview.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.nlp.CookingStackExchange.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.nlp.TextDataset.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.nlp.WikiText2.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Sets the optional artifactId.
optArtifactId(String) - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
Sets the optional artifactId.
optCsvFile(Path) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
Sets the optional CSV file path.
optCsvUrl(String) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
Sets the optional CSV file URL.
optDatasetName(String) - Method in class ai.djl.basicdataset.nlp.AmazonReview.Builder
Sets the name of the subset of Amazon Reviews.
optFlag(Image.Flag) - Method in class ai.djl.basicdataset.cv.ImageDataset.BaseBuilder
Sets the optional color mode flag.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.BananaDetection.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.classification.Cifar10.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.classification.FashionMnist.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.classification.Mnist.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.CocoDetection.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.cv.PikachuDetection.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.nlp.AmazonReview.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.nlp.CookingStackExchange.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.nlp.TextDataset.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.nlp.WikiText2.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Sets optional groupId.
optGroupId(String) - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
Sets optional groupId.
optImageHeight(int) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
Sets the height of the images.
optImageSize(int) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
Sets the size of the images.
optImageWidth(int) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
Sets the width of the images.
optManager(NDManager) - Method in class ai.djl.basicdataset.cv.classification.Cifar10.Builder
Sets the optional manager for the dataset (default follows engine default).
optManager(NDManager) - Method in class ai.djl.basicdataset.cv.classification.FashionMnist.Builder
Sets the optional manager for the dataset (default follows engine default).
optManager(NDManager) - Method in class ai.djl.basicdataset.cv.classification.Mnist.Builder
Sets the optional manager for the dataset (default follows engine default).
optManager(NDManager) - Method in class ai.djl.basicdataset.nlp.TextDataset.Builder
Sets the optional manager for the dataset (default follows engine default).
optMaxDepth(int) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
Sets the depth of the image folder.
optNormalize(boolean) - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Sets if normalize the dataset.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.BananaDetection.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.classification.Cifar10.Builder
Sets the optional repository for the dataset.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.classification.FashionMnist.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.classification.Mnist.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.CocoDetection.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.cv.PikachuDetection.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.nlp.AmazonReview.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.nlp.CookingStackExchange.Builder
Sets the optional repository for the dataset.
optRepository(Repository) - Method in class ai.djl.basicdataset.nlp.TextDataset.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.nlp.WikiText2.Builder
Sets the optional repository for the dataset.
optRepository(Repository) - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Sets the optional repository.
optRepository(Repository) - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
Sets the optional repository.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.BananaDetection.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.classification.Cifar10.Builder
Sets the optional usage for the dataset.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.classification.FashionMnist.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.classification.ImageNet.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.classification.Mnist.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.CocoDetection.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.cv.PikachuDetection.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.nlp.CookingStackExchange.Builder
Sets the optional usage for the dataset.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.nlp.TextDataset.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.nlp.WikiText2.Builder
Sets the optional usage for the dataset.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Sets the optional usage.
optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
Sets the optional usage.

P

PennTreebankText - Class in ai.djl.basicdataset.nlp
The Penn Treebank (PTB) project selected 2,499 stories from a three year Wall Street Journal (WSJ) collection of 98,732 stories for syntactic annotation (see here for details).
PennTreebankText.Builder - Class in ai.djl.basicdataset.nlp
A builder to construct a PennTreebankText .
PikachuDetection - Class in ai.djl.basicdataset.cv
Pikachu image detection dataset that contains multiple Pikachus in each image.
PikachuDetection(PikachuDetection.Builder) - Constructor for class ai.djl.basicdataset.cv.PikachuDetection
 
PikachuDetection.Builder - Class in ai.djl.basicdataset.cv
A builder for a PikachuDetection.
prepare() - Method in class ai.djl.basicdataset.cv.CocoUtils
Prepares and indexes the annotation file in memory.
prepare(Progress) - Method in class ai.djl.basicdataset.cv.BananaDetection
prepare(Progress) - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset
prepare(Progress) - Method in class ai.djl.basicdataset.cv.classification.Cifar10
prepare(Progress) - Method in class ai.djl.basicdataset.cv.classification.FashionMnist
prepare(Progress) - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten
prepare(Progress) - Method in class ai.djl.basicdataset.cv.classification.ImageFolder
prepare(Progress) - Method in class ai.djl.basicdataset.cv.classification.ImageNet
prepare(Progress) - Method in class ai.djl.basicdataset.cv.classification.Mnist
prepare(Progress) - Method in class ai.djl.basicdataset.cv.CocoDetection
prepare(Progress) - Method in class ai.djl.basicdataset.cv.PikachuDetection
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.AmazonReview
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.CookingStackExchange
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.GoEmotions
Prepares the dataset for use with tracked progress.
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.PennTreebankText
Prepares the dataset for use with tracked progress.
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.StanfordMovieReview
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset
Prepares the dataset for use with tracked progress.
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT
Prepares the dataset for use with tracked progress.
prepare(Progress) - Method in class ai.djl.basicdataset.nlp.WikiText2
Prepares the dataset for use with tracked progress.
prepare(Progress) - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess
prepare(Progress) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess
prepare(Progress) - Method in class ai.djl.basicdataset.tabular.CsvDataset
prepare(Progress) - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate
prepare(Progress) - Method in class ai.djl.basicdataset.tabular.MovieLens100k
prepare(List<String>) - Method in interface ai.djl.basicdataset.tabular.utils.PreparedFeaturizer
Prepares the featurizer with the list of possible inputs.
prepared - Variable in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
 
prepared - Variable in class ai.djl.basicdataset.nlp.TextDataset
 
PreparedFeaturizer - Interface in ai.djl.basicdataset.tabular.utils
A Featurizer that must be prepared with the possible feature values before use.
prepareFeaturizers() - Method in class ai.djl.basicdataset.tabular.TabularDataset
Prepares the PreparedFeaturizers.
preprocess(NDManager, List<String>) - Method in class ai.djl.basicdataset.utils.TextData
Preprocess the textData into NDArray by providing the data from the dataset.
preprocess(List<String>, boolean) - Method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset
Performs pre-processing steps on text data such as tokenising, applying TextProcessors, creating vocabulary, and word embeddings.
preprocess(List<String>, boolean) - Method in class ai.djl.basicdataset.nlp.TextDataset
Performs pre-processing steps on text data such as tokenising, applying TextProcessors, creating vocabulary, and word embeddings.
processInput(TranslatorContext, ListFeatures) - Method in class ai.djl.basicdataset.tabular.TabularTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.basicdataset.tabular.TabularTranslator
put(float) - Method in class ai.djl.basicdataset.tabular.utils.DynamicBuffer
Writes the given float into this buffer at the current position.

R

repository - Variable in class ai.djl.basicdataset.nlp.TextDataset.Builder
 
REPOSITORY - Static variable in interface ai.djl.basicdataset.BasicDatasets
 

S

sample(RandomAccessDataset) - Method in class ai.djl.basicdataset.utils.FixedBucketSampler
Sample(int, int) - Constructor for class ai.djl.basicdataset.nlp.TextDataset.Sample
Constructs a new Sample instance.
samples - Variable in class ai.djl.basicdataset.nlp.TextDataset
 
self() - Method in class ai.djl.basicdataset.cv.BananaDetection.Builder
self() - Method in class ai.djl.basicdataset.cv.classification.CaptchaDataset.Builder
self() - Method in class ai.djl.basicdataset.cv.classification.Cifar10.Builder
self() - Method in class ai.djl.basicdataset.cv.classification.FashionMnist.Builder
self() - Method in class ai.djl.basicdataset.cv.classification.FruitsFreshAndRotten.Builder
self() - Method in class ai.djl.basicdataset.cv.classification.ImageFolder.Builder
self() - Method in class ai.djl.basicdataset.cv.classification.ImageNet.Builder
self() - Method in class ai.djl.basicdataset.cv.classification.Mnist.Builder
self() - Method in class ai.djl.basicdataset.cv.CocoDetection.Builder
self() - Method in class ai.djl.basicdataset.cv.PikachuDetection.Builder
self() - Method in class ai.djl.basicdataset.nlp.AmazonReview.Builder
self() - Method in class ai.djl.basicdataset.nlp.GoEmotions.Builder
self() - Method in class ai.djl.basicdataset.nlp.PennTreebankText.Builder
self() - Method in class ai.djl.basicdataset.nlp.StanfordMovieReview.Builder
self() - Method in class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset.Builder
self() - Method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset.Builder
self() - Method in class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT.Builder
self() - Method in class ai.djl.basicdataset.tabular.AirfoilRandomAccess.Builder
self() - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
self() - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
self() - Method in class ai.djl.basicdataset.tabular.DailyDelhiClimate.Builder
Returns this {code Builder} object.
self() - Method in class ai.djl.basicdataset.tabular.MovieLens100k.Builder
setCsvFormat(CSVFormat) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
Sets the CSV file format.
setEmbeddingSize(int) - Method in class ai.djl.basicdataset.utils.TextData.Configuration
Sets the size for new TextEmbeddings.
setEmbeddingSize(int) - Method in class ai.djl.basicdataset.utils.TextData
Sets the embedding size.
setRepository(Repository) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
Sets the repository containing the image folder.
setRepositoryPath(String) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
Sets the repository file path containing the image folder.
setRepositoryPath(Path) - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder.ImageFolderBuilder
Sets the repository file path containing the image folder.
setReservedTokens(List<String>) - Method in class ai.djl.basicdataset.utils.TextData.Configuration
Sets the list of reserved tokens.
setSourceConfiguration(TextData.Configuration) - Method in class ai.djl.basicdataset.nlp.TextDataset.Builder
Sets the TextData.Configuration to use for the source text data.
setTargetConfiguration(TextData.Configuration) - Method in class ai.djl.basicdataset.nlp.TextDataset.Builder
Sets the TextData.Configuration to use for the target text data.
setTextEmbedding(TextEmbedding) - Method in class ai.djl.basicdataset.utils.TextData.Configuration
Sets the TextEmbedding to use to embed the text data.
setTextEmbedding(TextEmbedding) - Method in class ai.djl.basicdataset.utils.TextData
Sets the textEmbedding to embed the data with.
setTextProcessors(List<TextProcessor>) - Method in class ai.djl.basicdataset.utils.TextData.Configuration
Sets the TextProcessors to use for the text data.
setTextProcessors(List<TextProcessor>) - Method in class ai.djl.basicdataset.utils.TextData
Sets the text processors.
setUnknownToken(String) - Method in class ai.djl.basicdataset.utils.TextData.Configuration
Sets the default unknown token.
setVocabulary(Vocabulary) - Method in class ai.djl.basicdataset.utils.TextData.Configuration
Sets the Vocabulary to use to hold the text data.
size() - Method in class ai.djl.basicdataset.tabular.TabularResults
Returns the number of results.
sourceTextData - Variable in class ai.djl.basicdataset.nlp.TextDataset
 
StanfordMovieReview - Class in ai.djl.basicdataset.nlp
The StanfordMovieReview dataset contains a Application.NLP.SENTIMENT_ANALYSIS set of movie reviews and their sentiment ratings.
StanfordMovieReview(StanfordMovieReview.Builder) - Constructor for class ai.djl.basicdataset.nlp.StanfordMovieReview
Creates a new instance of StanfordMovieReview with the given necessary configurations.
StanfordMovieReview.Builder - Class in ai.djl.basicdataset.nlp
A builder for a StanfordMovieReview.
StanfordQuestionAnsweringDataset - Class in ai.djl.basicdataset.nlp
Stanford Question Answering Dataset (SQuAD) is a reading comprehension dataset, consisting of questions posed by crowdworkers on a set of Wikipedia articles, where the answer to every question is a segment of text, or span, from the corresponding reading passage, or the question might be unanswerable.
StanfordQuestionAnsweringDataset(StanfordQuestionAnsweringDataset.Builder) - Constructor for class ai.djl.basicdataset.nlp.StanfordQuestionAnsweringDataset
Creates a new instance of StanfordQuestionAnsweringDataset.
StanfordQuestionAnsweringDataset.Builder - Class in ai.djl.basicdataset.nlp
synset - Variable in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
 

T

TabularDataset - Class in ai.djl.basicdataset.tabular
A abstract class for creating tabular datasets.
TabularDataset(TabularDataset.BaseBuilder<?>) - Constructor for class ai.djl.basicdataset.tabular.TabularDataset
Creates a new instance of RandomAccessDataset with the given necessary configurations.
TabularDataset.BaseBuilder<T extends TabularDataset.BaseBuilder<T>> - Class in ai.djl.basicdataset.tabular
Used to build a TabularDataset.
TabularResult(String, Object) - Constructor for class ai.djl.basicdataset.tabular.TabularResults.TabularResult
Constructs the result.
TabularResults - Class in ai.djl.basicdataset.tabular
A list of results from running a tabular model.
TabularResults(List<TabularResults.TabularResult>) - Constructor for class ai.djl.basicdataset.tabular.TabularResults
Constructs a TabularResults with the given results.
TabularResults.TabularResult - Class in ai.djl.basicdataset.tabular
A single result corresponding to a single feature.
TabularTranslator - Class in ai.djl.basicdataset.tabular
A Translator that can be used for Application.Tabular tasks.
TabularTranslator(Model, Map<String, ?>) - Constructor for class ai.djl.basicdataset.tabular.TabularTranslator
Constructs a tabular translator for a model.
TabularTranslator(List<Feature>, List<Feature>) - Constructor for class ai.djl.basicdataset.tabular.TabularTranslator
Constructs a TabularTranslator with the given features and labels.
TabularTranslatorFactory - Class in ai.djl.basicdataset.tabular
A TranslatorFactory to extend the TabularTranslator.
TabularTranslatorFactory() - Constructor for class ai.djl.basicdataset.tabular.TabularTranslatorFactory
 
targetTextData - Variable in class ai.djl.basicdataset.nlp.TextDataset
 
TatoebaEnglishFrenchDataset - Class in ai.djl.basicdataset.nlp
TatoebaEnglishFrenchDataset is a English-French machine translation dataset from The Tatoeba Project (http://www.manythings.org/anki/).
TatoebaEnglishFrenchDataset(TatoebaEnglishFrenchDataset.Builder) - Constructor for class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset
Creates a new instance of TatoebaEnglishFrenchDataset.
TatoebaEnglishFrenchDataset.Builder - Class in ai.djl.basicdataset.nlp
TextData - Class in ai.djl.basicdataset.utils
TextData is a utility for managing textual data within a Dataset.
TextData(TextData.Configuration) - Constructor for class ai.djl.basicdataset.utils.TextData
Constructs a new TextData.
TextData.Configuration - Class in ai.djl.basicdataset.utils
The configuration for creating a TextData value in a Dataset.
TextDataset - Class in ai.djl.basicdataset.nlp
TextDataset is an abstract dataset that can be used for datasets for natural language processing where either the source or target are text-based data.
TextDataset(TextDataset.Builder<?>) - Constructor for class ai.djl.basicdataset.nlp.TextDataset
Creates a new instance of RandomAccessDataset with the given necessary configurations.
TextDataset.Builder<T extends TextDataset.Builder<T>> - Class in ai.djl.basicdataset.nlp
Abstract Builder that helps build a TextDataset.
TextDataset.Sample - Class in ai.djl.basicdataset.nlp
A class stores TextDataset sample information.
ThrowingFunction<T,​R,​E extends java.lang.Exception> - Interface in ai.djl.basicdataset.utils
Represents a function that accepts one argument, produces a result, and throws an Exception.

U

UniversalDependenciesEnglishEWT - Class in ai.djl.basicdataset.nlp
A Gold Standard Universal Dependencies Corpus for English, built over the source material of the English Web Treebank LDC2012T13.
UniversalDependenciesEnglishEWT(UniversalDependenciesEnglishEWT.Builder) - Constructor for class ai.djl.basicdataset.nlp.UniversalDependenciesEnglishEWT
Creates a new instance of UniversalDependenciesEnglish.
UniversalDependenciesEnglishEWT.Builder - Class in ai.djl.basicdataset.nlp
update(TextData.Configuration) - Method in class ai.djl.basicdataset.utils.TextData.Configuration
Updates this TextData.Configuration with the non-null values from another configuration.
usage - Variable in class ai.djl.basicdataset.nlp.TextDataset.Builder
 
usage - Variable in class ai.djl.basicdataset.nlp.TextDataset
 

W

WikiText2 - Class in ai.djl.basicdataset.nlp
The WikiText language modeling dataset is a collection of over 100 million tokens extracted from the set of verified Good and Featured articles on Wikipedia.
WikiText2.Builder - Class in ai.djl.basicdataset.nlp
A builder to construct a WikiText2 .
A B C D F G I L M N O P R S T U W 
All Classes All Packages