lamp-data/lamp.data/BatchStream

BatchStream

object BatchStream
Companion:
class
class Object
trait Matchable
class Any

Type members

Classlikes

object StagedLoader

Value members

Concrete methods

def fromFullBatch(features: STen, targets: STen, device: Device): BatchStream[Variable, Boolean, Unit]
def fromIndices[A, C](indices: Array[Array[Int]])(makeNonEmptyBatch: (Array[Int], Device) => Resource[IO, StreamControl[(A, STen)]]): BatchStream[A, Int, Unit]
def fromIndicesWithBuffers[A, C](indices: Array[Array[Int]], allocateBuffers1: Device => Resource[IO, C])(makeNonEmptyBatch: (Array[Int], C, Device) => Resource[IO, StreamControl[(A, STen)]]): BatchStream[A, Int, C]
def minibatchesFromFull(minibatchSize: Int, dropLast: Boolean, features: STen, target: STen, rng: Random): BatchStream[Variable, Int, BufferPair]
def single[A](resource: Resource[IO, StreamControl[(A, STen)]]): BatchStream[A, Boolean, Unit]
def stagedFromIndices[A, B, C](indices: Array[Array[Int]], bucketSize: Int, allocateBuffers0: Device => Resource[IO, C])(loadInstancesToStaging: Array[Int] => Resource[IO, B], makeNonEmptyBatch: (B, Array[Int], C, Device) => Resource[IO, StreamControl[(A, STen)]]): BatchStream[A, State[A, B], C]