- 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
-
- addCategoricalFeature(String) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
Adds a categorical feature to the feature set.
- addCategoricalFeature(String, Map<String, Integer>, boolean) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
Adds a categorical feature to the feature set with specified mapping.
- addCategoricalLabel(String) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
Adds a categorical feature to the label set.
- addCategoricalLabel(String, Map<String, Integer>, boolean) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
Adds a categorical feature to the feature set with specified mapping.
- 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, boolean) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
-
Adds a feature to the features set with onehot encoding.
- addFeature(CsvDataset.Feature...) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
Adds the features to the feature set.
- addLabel(CsvDataset.Feature...) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
Adds the features to the label set.
- addNumericFeature(String) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
Adds a numeric feature to the feature set.
- addNumericLabel(String) - Method in class ai.djl.basicdataset.tabular.CsvDataset.CsvBuilder
-
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.utils - package ai.djl.basicdataset.utils
-
Contains utilities used within the basic datasets.
- AirfoilRandomAccess - Class in ai.djl.basicdataset.tabular
-
Airfoil Self-Noise Data Set from https://archive.ics.uci.edu/ml/datasets/Airfoil+Self-Noise.
- AirfoilRandomAccess.Builder - Class in ai.djl.basicdataset.tabular
-
- 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
-
- 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.StanfordMovieReview
- availableSize() - Method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset
- availableSize() - Method in class ai.djl.basicdataset.tabular.CsvDataset
- 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.StanfordMovieReview
- get(NDManager, long) - Method in class ai.djl.basicdataset.nlp.TatoebaEnglishFrenchDataset
- get(NDManager, long) - Method in class ai.djl.basicdataset.tabular.CsvDataset
- getAnnotationById(long) - Method in class ai.djl.basicdataset.cv.CocoUtils
-
- 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.
- 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.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.
- 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.
- 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
-
Returns a plain java object.
- getData(NDManager) - Method in class ai.djl.basicdataset.nlp.CookingStackExchange
- getData() - Method in interface ai.djl.basicdataset.RawDataset
-
Returns a plain java object.
- getDefaultConfiguration() - Static method in class ai.djl.basicdataset.utils.TextData
-
- getEmbedding(NDManager, long) - Method in class ai.djl.basicdataset.utils.TextData
-
Gets the text embedding for the given index of the text input.
- getFeaturizer() - Method in class ai.djl.basicdataset.tabular.CsvDataset.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
-
- 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
-
- 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
-
- 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
-
- 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.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
-
- 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
-
- 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
-
- getIndex() - Method in class ai.djl.basicdataset.nlp.TextDataset.Sample
-
Returns the sample index.
- getLength() - Method in class ai.djl.basicdataset.utils.DynamicBuffer
-
Returns the buffer size.
- getName() - Method in class ai.djl.basicdataset.tabular.CsvDataset.Feature
-
Returns the feature name.
- getObjects(long) - Method in class ai.djl.basicdataset.cv.BananaDetection
-
- getObjects(long) - Method in class ai.djl.basicdataset.cv.CocoDetection
-
- 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
-
- getProcessedText(long, boolean) - Method in class ai.djl.basicdataset.nlp.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.nlp.TextDataset
-
Gets the raw textual input.
- getRawText(long) - Method in class ai.djl.basicdataset.utils.TextData
-
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.
- 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.
- getSynset() - Method in class ai.djl.basicdataset.cv.classification.AbstractImageFolder
-
Returns the synsets of the ImageFolder dataset.
- getTextEmbedding(boolean) - Method in class ai.djl.basicdataset.nlp.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.
- getVocabulary(boolean) - Method in class ai.djl.basicdataset.nlp.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.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.
- GROUP_ID - Static variable in interface ai.djl.basicdataset.BasicDatasets
-
- groupId - Variable in class ai.djl.basicdataset.nlp.TextDataset.Builder
-
- 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
-
- 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.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.tabular.AirfoilRandomAccess.Builder
-
Sets the optional artifactId.
- optArtifactId(String) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.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.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.tabular.AirfoilRandomAccess.Builder
-
Sets optional groupId.
- optGroupId(String) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.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.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.tabular.AirfoilRandomAccess.Builder
-
Sets the optional repository.
- optRepository(Repository) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.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.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.tabular.AirfoilRandomAccess.Builder
-
Sets the optional usage.
- optUsage(Dataset.Usage) - Method in class ai.djl.basicdataset.tabular.AmesRandomAccess.Builder
-
Sets the optional usage.