skainet-data-api/sk.ainet.data/Dataset

Dataset

abstract class Dataset<T, Y>(source)

Just abstract Dataset.

Constructors

constructor()

Types

inner class BatchIterator<T : DType, V> : Iterator<DataBatch<T, V>>

An iterator over a Dataset.

Properties

abstract val xSize: Int

Returns amount of data rows.

Functions

fun <T : DType, V> batchIterator(batchSize: Int): Dataset.BatchIterator<T, V>

Returns BatchIterator with fixed batchSize.

abstract fun getX(idx: Int): T

Returns row by index idx.

abstract fun getY(idx: Int): Y

Returns label as Int by index idx.

abstract fun shuffle(): Dataset<T, Y>

Shuffles the dataset.

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

Splits datasets on two sub-datasets according splitRatio.