mnist

Documentation

Index

• func Load(mode Mode, baseDir string) (inputs, targets tensor.Tensor, err error)
• type ContentType
• type Mode

Functions

func Load

func Load(mode Mode, baseDir string) (inputs, targets tensor.Tensor, err error)

Types

type ContentType

type ContentType string

const (
	ContentTypeData   ContentType = "data"
	ContentTypeLabels ContentType = "labels"
)

type Mode

type Mode string

const (
	ModeTrain Mode = "train"
	ModeTest  Mode = "test"
)