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, STen), Boolean, Unit]
def fromIndices[A, C](indices: Array[Array[Int]])(makeNonEmptyBatch: (Array[Int], Device) => Resource[IO, StreamControl[(A, STen)]]): BatchStream[(A, STen), Int, Unit]
def fromIndicesWithBuffers[A, C](indices: Array[Array[Int]], allocateBuffers1: Device => Resource[IO, C])(makeNonEmptyBatch: (Array[Int], C, Device) => Resource[IO, StreamControl[A]]): BatchStream[A, Int, C]
def fromVector[A](resources: Vector[Resource[IO, StreamControl[A]]]): BatchStream[A, Int, Unit]
def minibatchesFromFull(minibatchSize: Int, dropLast: Boolean, features: STen, target: STen, rng: Random): BatchStream[(Variable, STen), Int, BufferPair]
def single[A](resource: Resource[IO, StreamControl[A]]): 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]]): BatchStream[A, State[A, B], C]