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

buildStringRepr

Builds intermediate OnHeapDataset string representation.

fullToString

Creates OnHeapDataset string representation for full of data.

toString

Creates OnHeapDataset string representation for part of data.