Uses of Interface
io.github.kirstenali.deepj.data.BatchSource

Packages that use BatchSource

Package	Description
io.github.kirstenali.deepj.data
io.github.kirstenali.deepj.training

Uses of BatchSource in io.github.kirstenali.deepj.data

Subinterfaces of BatchSource in io.github.kirstenali.deepj.data

Modifier and Type	Interface	Description
interface	StatefulBatchSource
interface	StatefulTrainingDataset

Classes in io.github.kirstenali.deepj.data that implement BatchSource

Modifier and Type	Class	Description
final class	IndexedResponseTextDataset
final class	RandomAccessTextDataset
final class	ResponseOnlyTextDataset
final class	SequentialTextDataset
final class	TextDataset

Uses of BatchSource in io.github.kirstenali.deepj.training

Methods in io.github.kirstenali.deepj.training with parameters of type BatchSource

Modifier and Type	Method	Description
static EvaluationResult	CausalLMEvaluation.evaluate(CausalLM model, BatchSource source, int batches, int batchSize)
static Trainer	CausalLMTraining.trainer(CausalLM model, BatchSource dataset, float lr)
static Trainer	CausalLMTraining.trainer(CausalLM model, BatchSource dataset, ParameterOptimizer optimizer)