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

A

AbstractImageFolder - Class in ai.djl.basicdataset

A dataset for loading image files stored in a folder structure.

AbstractImageFolder(AbstractImageFolder.ImageFolderBuilder<?>) - Constructor for class ai.djl.basicdataset.AbstractImageFolder

AbstractImageFolder.ImageFolderBuilder<T extends AbstractImageFolder.ImageFolderBuilder> - Class in ai.djl.basicdataset

Used to build an AbstractImageFolder.

addAllFeatures() - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Move all available features to set.

addAllFeatures() - Method in class ai.djl.basicdataset.AmesRandomAccess

Move all non set features to be used.

addFeature(String) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Adds a feature if it exists.

addFeature(String, AmesRandomAccess.FeatureType) - Method in class ai.djl.basicdataset.AmesRandomAccess

Adds a feature of the given type to the used features set.

addFeature(String) - Method in class ai.djl.basicdataset.AmesRandomAccess

Adds a feature if it exists.

ai.djl.basicdataset - package ai.djl.basicdataset

Contains a library of built-in datasets.

ai.djl.basicdataset.utils - package ai.djl.basicdataset.utils

Contains utilities used within the basic datasets.

AirfoilRandomAccess - Class in ai.djl.basicdataset

Airfoil Self-Noise Data Set from https://archive.ics.uci.edu/ml/datasets/Airfoil+Self-Noise.

AirfoilRandomAccess.Builder - Class in ai.djl.basicdataset

A builder to construct a AirfoilRandomAccess.

AmesRandomAccess - Class in ai.djl.basicdataset

Ames house pricing dataset from https://www.kaggle.com/c/house-prices-advanced-regression-techniques/data.

AmesRandomAccess.Builder - Class in ai.djl.basicdataset

A builder to construct a AmesRandomAccess.

AmesRandomAccess.FeatureType - Enum in ai.djl.basicdataset

An enum represent data type of feature.

Annotation() - Constructor for class ai.djl.basicdataset.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.TextDataset.Builder

availableSize() - Method in class ai.djl.basicdataset.AbstractImageFolder

availableSize() - Method in class ai.djl.basicdataset.AirfoilRandomAccess

availableSize() - Method in class ai.djl.basicdataset.AmesRandomAccess

availableSize() - Method in class ai.djl.basicdataset.BananaDetection

availableSize() - Method in class ai.djl.basicdataset.CaptchaDataset

availableSize() - Method in class ai.djl.basicdataset.CocoDetection

availableSize() - Method in class ai.djl.basicdataset.PikachuDetection

availableSize() - Method in class ai.djl.basicdataset.StanfordMovieReview

availableSize() - Method in class ai.djl.basicdataset.TatoebaEnglishFrenchDataset

B

BananaDetection - Class in ai.djl.basicdataset

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

Creates a new instance of RandomAccessDataset with the given necessary configurations.

BananaDetection.Builder - Class in ai.djl.basicdataset

A builder for a BananaDetection.

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.AirfoilRandomAccess.Builder

Builds the new AirfoilRandomAccess.

build() - Method in class ai.djl.basicdataset.AmesRandomAccess.Builder

Builds the new AmesRandomAccess.

build() - Method in class ai.djl.basicdataset.BananaDetection.Builder

Builds the BananaDetection.

build() - Method in class ai.djl.basicdataset.CaptchaDataset.Builder

Builds the CaptchaDataset.

build() - Method in class ai.djl.basicdataset.Cifar10.Builder

Builds a new Cifar10.

build() - Method in class ai.djl.basicdataset.CocoDetection.Builder

Builds the new CocoDetection.

build() - Method in class ai.djl.basicdataset.CookingStackExchange.Builder

Builds a new CookingStackExchange.

build() - Method in class ai.djl.basicdataset.FashionMnist.Builder

Builds the Mnist.

build() - Method in class ai.djl.basicdataset.ImageFolder.Builder

Builds the ImageFolder.

build() - Method in class ai.djl.basicdataset.ImageNet.Builder

Builds the ImageNet.

build() - Method in class ai.djl.basicdataset.Mnist.Builder

Builds the Mnist.

build() - Method in class ai.djl.basicdataset.PikachuDetection.Builder

Builds the PikachuDetection.

build() - Method in class ai.djl.basicdataset.StanfordMovieReview.Builder

Builds the StanfordMovieReview.

build() - Method in class ai.djl.basicdataset.TatoebaEnglishFrenchDataset.Builder

Builds the TatoebaEnglishFrenchDataset.

builder() - Static method in class ai.djl.basicdataset.AirfoilRandomAccess

Creates a builder to build a AirfoilRandomAccess.

builder() - Static method in class ai.djl.basicdataset.AmesRandomAccess

Creates a builder to build a AmesRandomAccess.

builder() - Static method in class ai.djl.basicdataset.BananaDetection

Creates a new builder to build a BananaDetection.

builder() - Static method in class ai.djl.basicdataset.CaptchaDataset

Creates a builder to build a CaptchaDataset.

builder() - Static method in class ai.djl.basicdataset.Cifar10

Creates a builder to build a Cifar10.

builder() - Static method in class ai.djl.basicdataset.CocoDetection

Creates a builder to build a CocoDetection.

builder() - Static method in class ai.djl.basicdataset.CookingStackExchange

Creates a builder to build a CookingStackExchange.

builder() - Static method in class ai.djl.basicdataset.FashionMnist

Creates a builder to build a Mnist.

builder() - Static method in class ai.djl.basicdataset.ImageFolder

Creates a new builder to build a ImageFolder.

builder() - Static method in class ai.djl.basicdataset.ImageNet

Creates a new builder to build a ImageNet.

builder() - Static method in class ai.djl.basicdataset.Mnist

Creates a builder to build a Mnist.

builder() - Static method in class ai.djl.basicdataset.PikachuDetection

Creates a new builder to build a PikachuDetection.

builder() - Static method in class ai.djl.basicdataset.StanfordMovieReview

Creates a new builder to build a StanfordMovieReview.

Builder() - Constructor for class ai.djl.basicdataset.StanfordMovieReview.Builder

Constructs a new builder.

builder() - Static method in class ai.djl.basicdataset.TatoebaEnglishFrenchDataset

Creates a new builder to build a TatoebaEnglishFrenchDataset.

Builder() - Constructor for class ai.djl.basicdataset.TatoebaEnglishFrenchDataset.Builder

Constructs a new builder.

C

CAPTCHA_LENGTH - Static variable in class ai.djl.basicdataset.CaptchaDataset

CAPTCHA_OPTIONS - Static variable in class ai.djl.basicdataset.CaptchaDataset

CaptchaDataset - Class in ai.djl.basicdataset

A Dataset featuring captcha images.

CaptchaDataset(CaptchaDataset.Builder) - Constructor for class ai.djl.basicdataset.CaptchaDataset

Creates a new instance of CaptchaDataset.

CaptchaDataset.Builder - Class in ai.djl.basicdataset

A builder for a CaptchaDataset.

Category() - Constructor for class ai.djl.basicdataset.CocoMetadata.Category

Cifar10 - Class in ai.djl.basicdataset

CIFAR10 image classification dataset from https://www.cs.toronto.edu/~kriz/cifar.html.

Cifar10.Builder - Class in ai.djl.basicdataset

A builder to construct a Cifar10.

CocoDetection - Class in ai.djl.basicdataset

Coco image detection dataset from http://cocodataset.org/#home.

CocoDetection.Builder - Class in ai.djl.basicdataset

A builder to construct a CocoDetection.

CocoMetadata - Class in ai.djl.basicdataset

A metadata class to represent the structure of annotations in Coco.

CocoMetadata() - Constructor for class ai.djl.basicdataset.CocoMetadata

CocoMetadata.Annotation - Class in ai.djl.basicdataset

An annotation applied to an image in the coco dataset.

CocoMetadata.Category - Class in ai.djl.basicdataset

An annotation category in the coco dataset.

CocoMetadata.Image - Class in ai.djl.basicdataset

An image in the coco dataset.

CocoMetadata.RectangleDeserializer - Class in ai.djl.basicdataset

A JsonDeserializer for the Rectangle class.

CocoUtils - Class in ai.djl.basicdataset

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

A text classification dataset contains questions from cooking.stackexchange.com and their associated tags on the site.

CookingStackExchange.Builder - Class in ai.djl.basicdataset

A builder to construct a CookingStackExchange.

D

deserialize(JsonElement, Type, JsonDeserializationContext) - Method in class ai.djl.basicdataset.CocoMetadata.RectangleDeserializer

DJL_REPO_URL - Static variable in interface ai.djl.basicdataset.BasicDatasets

F

FashionMnist - Class in ai.djl.basicdataset

FashMnist is a dataset from Zalando article images https://github.com/zalandoresearch/fashion-mnist.

FashionMnist.Builder - Class in ai.djl.basicdataset

A builder for a FashionMnist.

FixedBucketSampler - Class in ai.djl.basicdataset.utils

FixedBucketSampler is a Sampler to be used with TextDataset, and PaddingStackBatchifier.

FixedBucketSampler(int, int, boolean, 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.

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) - 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.AbstractImageFolder

G

get(NDManager, long) - Method in class ai.djl.basicdataset.AbstractImageFolder

get(NDManager, long) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

get(NDManager, long) - Method in class ai.djl.basicdataset.AmesRandomAccess

get(NDManager, long) - Method in class ai.djl.basicdataset.BananaDetection

get(NDManager, long) - Method in class ai.djl.basicdataset.CaptchaDataset

get(NDManager, long) - Method in class ai.djl.basicdataset.CocoDetection

get(NDManager, long) - Method in class ai.djl.basicdataset.PikachuDetection

get(NDManager, long) - Method in class ai.djl.basicdataset.StanfordMovieReview

get(NDManager, long) - Method in class ai.djl.basicdataset.TatoebaEnglishFrenchDataset

getAnnotationById(long) - Method in class ai.djl.basicdataset.CocoUtils

Returns an CocoMetadata.Annotation that corresponds to a given annotation id.

getAnnotationIdByImageId(long) - Method in class ai.djl.basicdataset.CocoUtils

Returns all ids of the annotation that correspond to a given image id.

getAnnotations() - Method in class ai.djl.basicdataset.CocoMetadata

Returns a list of all annotations.

getArea() - Method in class ai.djl.basicdataset.CocoMetadata.Annotation

Returns the area of this annotation.

getBatchSize() - Method in class ai.djl.basicdataset.utils.FixedBucketSampler

getBoundingBox() - Method in class ai.djl.basicdataset.CocoMetadata.Annotation

Returns the bounding box of this annotation.

getCategoricalFeatures() - Method in class ai.djl.basicdataset.AmesRandomAccess

Get categorical feature set.

getCategories() - Method in class ai.djl.basicdataset.CocoMetadata

Returns a list of all categories.

getCategoryId() - Method in class ai.djl.basicdataset.CocoMetadata.Annotation

Returns the category id of this annotation.

getClassFull() - Method in class ai.djl.basicdataset.ImageNet

Returns all full class names of this ImageNet dataset.

getClassNames() - Method in class ai.djl.basicdataset.ImageNet

Returns all class names of this ImageNet dataset.

getCocoUrl() - Method in class ai.djl.basicdataset.CocoMetadata.Image

Returns the url of this image.

getCSVRecord(int) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Returns the CSVRecord for a given index.

getCSVRecord(int) - Method in class ai.djl.basicdataset.AmesRandomAccess

Returns the CSVRecord for a given index.

getData() - Method in class ai.djl.basicdataset.CookingStackExchange

Returns a plain java object.

getData(NDManager) - Method in class ai.djl.basicdataset.CookingStackExchange

getData() - Method in interface ai.djl.basicdataset.RawDataset

Returns a plain java object.

getDefaultConfiguration() - Static method in class ai.djl.basicdataset.utils.TextData

Returns a good default TextData.Configuration to use for the constructor with defaults.

getDisabledFeatures() - Method in class ai.djl.basicdataset.AmesRandomAccess

Get disabled feature set.

getEmbedding(NDManager, long) - Method in class ai.djl.basicdataset.utils.TextData

Gets the text embedding for the given index of the text input.

getEnabledFeatures() - Method in class ai.djl.basicdataset.AmesRandomAccess

Get enabled feature set.

getFeatureArraySize() - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Returns the size of the feature array(column count).

getFeatureArraySize() - Method in class ai.djl.basicdataset.AmesRandomAccess

Returns the size of the feature array(column count).

getFeatureNDArray(NDManager, int) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Return the NDArray at index 'index' with the set features.

getFeatureNDArray(NDManager, int) - Method in class ai.djl.basicdataset.AmesRandomAccess

Return the NDArray at index 'index' with the set features.

getFeatureOrder() - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Gets the feature order of the columns.

getFeatureType(String) - Method in class ai.djl.basicdataset.AmesRandomAccess

Returns the feature type.

getHeight() - Method in class ai.djl.basicdataset.CocoMetadata.Image

Returns the height of this image.

getId() - Method in class ai.djl.basicdataset.CocoMetadata.Annotation

Returns the id of this annotation.

getId() - Method in class ai.djl.basicdataset.CocoMetadata.Category

Returns the id of this category.

getId() - Method in class ai.djl.basicdataset.CocoMetadata.Image

Returns the id of this image.

getImageId() - Method in class ai.djl.basicdataset.CocoMetadata.Annotation

Returns the id of the image this annotation applies to.

getImageIds() - Method in class ai.djl.basicdataset.CocoUtils

Returns all image ids in the annotation file.

getImagePath(String) - Method in class ai.djl.basicdataset.AbstractImageFolder

getImagePath(String) - Method in class ai.djl.basicdataset.ImageFolder

getImagePath(String) - Method in class ai.djl.basicdataset.ImageNet

getImages() - Method in class ai.djl.basicdataset.CocoMetadata

Returns a list of all images.

getLabel(int) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Returns the label value for a given index.

getLabel(int) - Method in class ai.djl.basicdataset.AmesRandomAccess

Returns the label value for a given index.

getProcessedText(long, boolean) - Method in class ai.djl.basicdataset.TextDataset

Gets the processed textual input.

getProcessedText(long) - Method in class ai.djl.basicdataset.utils.TextData

Gets the textual input after preprocessing.

getRawText(long, boolean) - Method in class ai.djl.basicdataset.TextDataset

Gets the raw textual input.

getRawText(long) - Method in class ai.djl.basicdataset.utils.TextData

Gets the raw textual input.

getRecordFloat(CSVRecord, String) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Returns float for a given record and feature.

getRelativeImagePath(long) - Method in class ai.djl.basicdataset.CocoUtils

Returns the relative path of an image given an image id.

getSize() - Method in class ai.djl.basicdataset.utils.TextData

Returns the size of the data.

getSynset() - Method in class ai.djl.basicdataset.AbstractImageFolder

Returns the synsets of the ImageFolder dataset.

getTextEmbedding(boolean) - Method in class ai.djl.basicdataset.TextDataset

Gets the word embedding used while pre-processing the dataset.

getTextEmbedding() - Method in class ai.djl.basicdataset.utils.TextData

Gets the TextEmbedding used to embed the data with.

getValueFloat(CSVRecord, String) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Returns the float value of the record's feature in a float[].

getValueFloat(CSVRecord, String) - Method in class ai.djl.basicdataset.AmesRandomAccess

Returns the float value of the record's feature Changes based on categorical or numeric and for categorical if 1 hot encode is enabled.

getVocabulary(boolean) - Method in class ai.djl.basicdataset.TextDataset

Gets the SimpleVocabulary built while preprocessing the text data.

getVocabulary() - Method in class ai.djl.basicdataset.utils.TextData

Gets the SimpleVocabulary built while preprocessing the text data.

getWidth() - Method in class ai.djl.basicdataset.CocoMetadata.Image

Returns the width of this image.

getWordNetIds() - Method in class ai.djl.basicdataset.ImageNet

Returns all WordNet ids of this ImageNet dataset.

GROUP_ID - Static variable in interface ai.djl.basicdataset.BasicDatasets

groupId - Variable in class ai.djl.basicdataset.TextDataset.Builder

I

Image() - Constructor for class ai.djl.basicdataset.CocoMetadata.Image

IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.CaptchaDataset

IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.Cifar10

IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.FashionMnist

IMAGE_HEIGHT - Static variable in class ai.djl.basicdataset.Mnist

IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.CaptchaDataset

IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.Cifar10

IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.FashionMnist

IMAGE_WIDTH - Static variable in class ai.djl.basicdataset.Mnist

ImageFolder - Class in ai.djl.basicdataset

A dataset for loading image files stored in a folder structure.

ImageFolder.Builder - Class in ai.djl.basicdataset

A builder for the ImageFolder.

ImageFolderBuilder() - Constructor for class ai.djl.basicdataset.AbstractImageFolder.ImageFolderBuilder

ImageNet - Class in ai.djl.basicdataset

ImageNet is an image classification dataset from http://image-net.org 2012 Classification dataset.

ImageNet.Builder - Class in ai.djl.basicdataset

A builder for a ImageNet.

items - Variable in class ai.djl.basicdataset.AbstractImageFolder

L

listImages(Path, List<String>) - Method in class ai.djl.basicdataset.AbstractImageFolder

M

manager - Variable in class ai.djl.basicdataset.TextDataset

mapCategoryId(long) - Method in class ai.djl.basicdataset.CocoUtils

Returns the continuous category id given an original category id.

Mnist - Class in ai.djl.basicdataset

MNIST handwritten digits dataset from http://yann.lecun.com/exdb/mnist.

Mnist.Builder - Class in ai.djl.basicdataset

A builder for a Mnist.

N

NORMALIZE_MEAN - Static variable in class ai.djl.basicdataset.Cifar10

NORMALIZE_STD - Static variable in class ai.djl.basicdataset.Cifar10

NUM_CLASSES - Static variable in class ai.djl.basicdataset.FashionMnist

NUM_CLASSES - Static variable in class ai.djl.basicdataset.Mnist

O

optArtifactId(String) - Method in class ai.djl.basicdataset.AirfoilRandomAccess.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.AmesRandomAccess.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.BananaDetection.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.CaptchaDataset.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.Cifar10.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.CocoDetection.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.CookingStackExchange.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.FashionMnist.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.Mnist.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.PikachuDetection.Builder

Sets the optional artifactId.

optArtifactId(String) - Method in class ai.djl.basicdataset.TextDataset.Builder

Sets the optional artifactId.

optFlag(Image.Flag) - Method in class ai.djl.basicdataset.AbstractImageFolder.ImageFolderBuilder

Sets the optional color mode flag.

optFlag(Image.Flag) - Method in class ai.djl.basicdataset.BananaDetection.Builder

Sets the optional color mode flag.

optFlag(Image.Flag) - Method in class ai.djl.basicdataset.CocoDetection.Builder

Sets the optional color mode flag.

optFlag(Image.Flag) - Method in class ai.djl.basicdataset.PikachuDetection.Builder

Sets the optional color mode flag.

optGroupId(String) - Method in class ai.djl.basicdataset.AirfoilRandomAccess.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.AmesRandomAccess.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.BananaDetection.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.CaptchaDataset.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.Cifar10.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.CocoDetection.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.CookingStackExchange.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.FashionMnist.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.Mnist.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.PikachuDetection.Builder

Sets optional groupId.

optGroupId(String) - Method in class ai.djl.basicdataset.TextDataset.Builder

Sets optional groupId.

optManager(NDManager) - Method in class ai.djl.basicdataset.Cifar10.Builder

Sets the optional manager for the dataset (default follows engine default).

optManager(NDManager) - Method in class ai.djl.basicdataset.FashionMnist.Builder

Sets the optional manager for the dataset (default follows engine default).

optManager(NDManager) - Method in class ai.djl.basicdataset.Mnist.Builder

Sets the optional manager for the dataset (default follows engine default).

optManager(NDManager) - Method in class ai.djl.basicdataset.TextDataset.Builder

Sets the optional manager for the dataset (default follows engine default).

optMaxDepth(int) - Method in class ai.djl.basicdataset.AbstractImageFolder.ImageFolderBuilder

Sets the depth of the image folder.

optRepository(Repository) - Method in class ai.djl.basicdataset.AirfoilRandomAccess.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.AmesRandomAccess.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.BananaDetection.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.CaptchaDataset.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.Cifar10.Builder

Sets the optional repository for the dataset.

optRepository(Repository) - Method in class ai.djl.basicdataset.CocoDetection.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.CookingStackExchange.Builder

Sets the optional repository for the dataset.

optRepository(Repository) - Method in class ai.djl.basicdataset.FashionMnist.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.Mnist.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.PikachuDetection.Builder

Sets the optional repository.

optRepository(Repository) - Method in class ai.djl.basicdataset.TextDataset.Builder

Sets the optional repository.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.AirfoilRandomAccess.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.AmesRandomAccess.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.BananaDetection.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.CaptchaDataset.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.Cifar10.Builder

Sets the optional usage for the dataset.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.CocoDetection.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.CookingStackExchange.Builder

Sets the optional usage for the dataset.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.FashionMnist.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.ImageNet.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.Mnist.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.PikachuDetection.Builder

Sets the optional usage.

optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.TextDataset.Builder

Sets the optional usage.

P

PikachuDetection - Class in ai.djl.basicdataset

Pikachu image detection dataset that contains multiple Pikachus in each image.

PikachuDetection(PikachuDetection.Builder) - Constructor for class ai.djl.basicdataset.PikachuDetection

PikachuDetection.Builder - Class in ai.djl.basicdataset

A builder for a PikachuDetection.

prepare(Progress) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

prepare(Progress) - Method in class ai.djl.basicdataset.AmesRandomAccess

prepare(Progress) - Method in class ai.djl.basicdataset.BananaDetection

prepare(Progress) - Method in class ai.djl.basicdataset.CaptchaDataset

prepare(Progress) - Method in class ai.djl.basicdataset.Cifar10

prepare(Progress) - Method in class ai.djl.basicdataset.CocoDetection

prepare() - Method in class ai.djl.basicdataset.CocoUtils

Prepares and indexes the annotation file in memory.

prepare(Progress) - Method in class ai.djl.basicdataset.CookingStackExchange

prepare(Progress) - Method in class ai.djl.basicdataset.FashionMnist

prepare(Progress) - Method in class ai.djl.basicdataset.ImageFolder

prepare(Progress) - Method in class ai.djl.basicdataset.ImageNet

prepare(Progress) - Method in class ai.djl.basicdataset.Mnist

prepare(Progress) - Method in class ai.djl.basicdataset.PikachuDetection

prepare(Progress) - Method in class ai.djl.basicdataset.StanfordMovieReview

prepare(Progress) - Method in class ai.djl.basicdataset.TatoebaEnglishFrenchDataset

prepared - Variable in class ai.djl.basicdataset.AbstractImageFolder

prepared - Variable in class ai.djl.basicdataset.TextDataset

preprocess(List<String>, boolean) - Method in class ai.djl.basicdataset.TextDataset

Performs pre-processing steps on text data such as tokenising, applying TextProcessors, creating vocabulary, and word embeddings.

preprocess(NDManager, List<String>) - Method in class ai.djl.basicdataset.utils.TextData

Preprocess the textData into NDArray by providing the data from the dataset.

R

RawDataset<T> - Interface in ai.djl.basicdataset

An interface can read a plain java object from dataset.

RectangleDeserializer() - Constructor for class ai.djl.basicdataset.CocoMetadata.RectangleDeserializer

removeAllFeatures() - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Move all currently set features to available.

removeAllFeatures() - Method in class ai.djl.basicdataset.AmesRandomAccess

Move all currently set features to available.

removeFeature(String) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Removes a feature from the active set if it exists.

removeFeature(String) - Method in class ai.djl.basicdataset.AmesRandomAccess

Removes a feature from the active set if it exists.

REPOSITORY - Static variable in interface ai.djl.basicdataset.BasicDatasets

repository - Variable in class ai.djl.basicdataset.TextDataset.Builder

resource - Variable in class ai.djl.basicdataset.AbstractImageFolder

resource - Variable in class ai.djl.basicdataset.TextDataset

S

sample(RandomAccessDataset) - Method in class ai.djl.basicdataset.utils.FixedBucketSampler

selectFirstN(int) - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Chooses the 1st N records to be used (not reversible).

self() - Method in class ai.djl.basicdataset.AirfoilRandomAccess.Builder

self() - Method in class ai.djl.basicdataset.AmesRandomAccess.Builder

self() - Method in class ai.djl.basicdataset.BananaDetection.Builder

self() - Method in class ai.djl.basicdataset.CaptchaDataset.Builder

self() - Method in class ai.djl.basicdataset.Cifar10.Builder

self() - Method in class ai.djl.basicdataset.CocoDetection.Builder

self() - Method in class ai.djl.basicdataset.FashionMnist.Builder

self() - Method in class ai.djl.basicdataset.ImageFolder.Builder

self() - Method in class ai.djl.basicdataset.ImageNet.Builder

self() - Method in class ai.djl.basicdataset.Mnist.Builder

self() - Method in class ai.djl.basicdataset.PikachuDetection.Builder

self() - Method in class ai.djl.basicdataset.StanfordMovieReview.Builder

self() - Method in class ai.djl.basicdataset.TatoebaEnglishFrenchDataset.Builder

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.

setFeatureType(String, AmesRandomAccess.FeatureType) - Method in class ai.djl.basicdataset.AmesRandomAccess

Sets a feature's type.

setLabel(String) - Method in class ai.djl.basicdataset.AmesRandomAccess

Sets the label(y).

setOneHotEncode(String, boolean) - Method in class ai.djl.basicdataset.AmesRandomAccess

Sets a feature's oneHotEncode status if it is categorical.

setRepository(Repository) - Method in class ai.djl.basicdataset.AbstractImageFolder.ImageFolderBuilder

Sets the repository containing the image folder.

setRepositoryPath(String) - Method in class ai.djl.basicdataset.AbstractImageFolder.ImageFolderBuilder

Sets the repository file path containing the image folder.

setRepositoryPath(Path) - Method in class ai.djl.basicdataset.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.TextDataset.Builder

Sets the TextData.Configuration to use for the source text data.

setTargetConfiguration(TextData.Configuration) - Method in class ai.djl.basicdataset.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.

sourceTextData - Variable in class ai.djl.basicdataset.TextDataset

StanfordMovieReview - Class in ai.djl.basicdataset

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

Creates a new instance of StanfordMovieReview with the given necessary configurations.

StanfordMovieReview.Builder - Class in ai.djl.basicdataset

A builder for a StanfordMovieReview.

synset - Variable in class ai.djl.basicdataset.AbstractImageFolder

T

targetTextData - Variable in class ai.djl.basicdataset.TextDataset

TatoebaEnglishFrenchDataset - Class in ai.djl.basicdataset

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

Creates a new instance of TatoebaEnglishFrenchDataset.

TatoebaEnglishFrenchDataset.Builder - Class in ai.djl.basicdataset

A builder for a TatoebaEnglishFrenchDataset.

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

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

Creates a new instance of RandomAccessDataset with the given necessary configurations.

TextDataset.Builder<T extends TextDataset.Builder<T>> - Class in ai.djl.basicdataset

Abstract Builder that helps build a TextDataset.

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

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.TextDataset.Builder

usage - Variable in class ai.djl.basicdataset.TextDataset

V

valueOf(String) - Static method in enum ai.djl.basicdataset.AmesRandomAccess.FeatureType

Returns the enum constant of this type with the specified name.

values() - Static method in enum ai.djl.basicdataset.AmesRandomAccess.FeatureType

Returns an array containing the constants of this enum type, in the order they are declared.

W

whitenAll() - Method in class ai.djl.basicdataset.AirfoilRandomAccess

Remove mean and rescale variance to 1 for all features.

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