Dataset

abstract class Dataset
Content copied to clipboard

Just abstract Dataset.

Constructors

Dataset
Link copied to clipboard
fun Dataset()
Content copied to clipboard

Types

BatchIterator
Link copied to clipboard
inner class BatchIterator : Iterator<DataBatch?>
Content copied to clipboard

An iterator over a Dataset.

Functions

batchIterator
Link copied to clipboard
fun batchIterator(batchSize: Int): Dataset.BatchIterator
Content copied to clipboard

Returns BatchIterator with fixed batchSize.

getX
Link copied to clipboard
abstract fun getX(idx: Int): FloatArray
Content copied to clipboard

Returns row by index idx.

getY
Link copied to clipboard
abstract fun getY(idx: Int): Float
Content copied to clipboard

Returns label as Int by index idx.

shuffle
Link copied to clipboard
abstract fun shuffle(): Dataset
Content copied to clipboard

Shuffles the dataset.

split
Link copied to clipboard
abstract fun split(splitRatio: Double): Pair<Dataset, Dataset>
Content copied to clipboard

Splits datasets on two sub-datasets according splitRatio.

xSize
Link copied to clipboard
abstract fun xSize(): Int
Content copied to clipboard

Returns amount of data rows.

Inheritors

OnFlyImageDataset
Link copied to clipboard
OnHeapDataset
Link copied to clipboard