CIFAR10Dataset

@Serializable

data class CIFAR10Dataset(val images: List<CIFAR10Image>, executionContext: ExecutionContext = DefaultDataExecutionContext()) : Dataset<CIFAR10Image, Float> (source)

CIFAR-10 dataset implementation using Dataset/DataBatch API.

Provides batching as tensors Int8 with shapes: x: batch, 3, 32, 32 (channel-first RGB) y: batch (labels as bytes)

Constructors

constructor(images: List<CIFAR10Image>, executionContext: ExecutionContext = DefaultDataExecutionContext())

open override val xSize: Int

open override fun getX(idx: Int): CIFAR10Image

open override fun getY(idx: Int): Float

fun subset(fromIndex: Int, toIndex: Int): CIFAR10Dataset

Returns a subset of the dataset.