OnHeapDataset

Basic class to handle features x and labels y.

It loads the whole data from disk to the Heap Memory.

NOTE: Labels y should have shape and contain exactly one 1 and other 0-es per row to be result of one-hot-encoding.

Types

Companion

batchIterator

getX

open override fun getX(idx: Int): FloatArray

Returns row by index idx.

getY

open override fun getY(idx: Int): Float

Returns label as FloatArray by index idx.

shuffle

open override fun shuffle(): OnHeapDataset

Shuffles the dataset.

split

Splits datasets on two sub-datasets according splitRatio.

xSize

open override fun xSize(): Int

Returns amount of data rows.

an array of feature vectors

an array of labels