Package org.jetbrains.kotlinx.dl.dataset

Functions

buildStringRepr
Link copied to clipboard
fun OnHeapDataset.buildStringRepr(xString: String, yString: String): String

Builds intermediate OnHeapDataset string representation.

cifar10Paths
Link copied to clipboard
fun cifar10Paths(cacheDirectory: File = File("cache")): Pair<String, String>

Returns paths to images and its labels for the Cifar'10 dataset.

dogsCatsDatasetPath
Link copied to clipboard
fun dogsCatsDatasetPath(cacheDirectory: File = File("cache")): String

Returns path to images of the Dogs-vs-Cats dataset.

dogsCatsSmallDatasetPath
Link copied to clipboard
fun dogsCatsSmallDatasetPath(cacheDirectory: File = File("cache")): String

Returns path to images of the subset of the Dogs-vs-Cats dataset.

fashionMnist
Link copied to clipboard
fun fashionMnist(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>

Loads the Fashion-MNIST dataset.

freeSpokenDigitDatasetPath
Link copied to clipboard
fun freeSpokenDigitDatasetPath(cacheDirectory: File = File("cache")): String

Returns path to sound data files from Free Spoken Digits Dataset.

freeSpokenDigits
Link copied to clipboard
fun freeSpokenDigits(cacheDirectory: File = File("cache"), maxTestIndex: Int = 5): Pair<OnHeapDataset, OnHeapDataset>

Loads the Free Spoken Digits Dataset. This is a dataset of wav sound files of the 10 digits spoken by different people many times each. The test set officially consists of the first 10% of the recordings. Recordings numbered 0-4 (inclusive) are in the test and 5-49 are in the training set.

fullToString
Link copied to clipboard
fun OnHeapDataset.fullToString(): String

Creates OnHeapDataset string representation for full of data.

mnist
Link copied to clipboard
fun mnist(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>

Loads the MNIST dataset. This is a dataset of 60,000 28x28 grayscale images of the 10 digits, along with a test set of 10,000 images. More info can be found at the MNIST homepage.

mnist3D
Link copied to clipboard
fun mnist3D(cacheDirectory: File = File("cache")): Pair<OnHeapDataset, OnHeapDataset>

Loads the MNIST 3D dataset. This is a dataset of 10,000 16x16x16 grayscale 3D images of the 10 digits, along with a test set of 2,000 3D images.

toString
Link copied to clipboard
fun OnHeapDataset.toString(): String

Creates OnHeapDataset string representation for part of data.

Properties

FSDD_SOUND_DATA_SIZE
Link copied to clipboard
const val FSDD_SOUND_DATA_SIZE: Long = 20480
MNIST_3D_DATASET
Link copied to clipboard
const val MNIST_3D_DATASET: String

Path to H5 file of Mnist 3D Dataset.