data

type Batch struct {
	X []*tf.Tensor
	Y *tf.Tensor
	// TODO: change class weights to a single tensor
	PosWeight *tf.Tensor
	NegWeight *tf.Tensor
}

type Dataset interface {
	Len() int
	SetMode(mode GeneratorMode) Dataset
	Shuffle(seed int64)
	Unshuffle() error
	GetColumnNames() []string
	GeneratorChan(batchSize int, preFetch int) chan Batch
	Generate(batchSize int) ([]*tf.Tensor, *tf.Tensor, error)
	Reset() error
}

type GeneratorMode string

type SingleFileDataset struct {
	ClassCounts map[int]int
	Count       int
	// contains filtered or unexported fields
}