Dataset

abstract class Dataset

Just abstract Dataset.

Constructors

Dataset

Types

BatchIterator

inner class BatchIterator : Iterator<DataBatch?>

An iterator over a Dataset.

Functions

batchIterator

fun batchIterator(batchSize: Int): Dataset.BatchIterator

Returns BatchIterator with fixed batchSize.

getX

abstract fun getX(idx: Int): FloatArray

Returns row by index idx.

getY

abstract fun getY(idx: Int): Float

Returns label as Int by index idx.

shuffle

abstract fun shuffle(): Dataset

Shuffles the dataset.

split

abstract fun split(splitRatio: Double): Pair<Dataset, Dataset>

Splits datasets on two sub-datasets according splitRatio.

xSize

abstract fun xSize(): Int

Returns amount of data rows.

Inheritors

OnFlyImageDataset

OnHeapDataset