mnist

package

v0.1.7 Latest Latest Go to latest Published: Mar 12, 2025 License: Apache-2.0 Imports: 8 Imported by: 2

Details

Package MNIST is the 60000 + 10000 handwritten digits dataset

View Source

const ImgSize = 28

original

View Source

const SmallImgSize = 13

downscaled

This section is empty.

func New() (TrainSet []Input, InferSet []Input, SmallTrainSet []SmallInput, SmallInferSet []SmallInput, err error)

type Input struct {
	Image [ImgSize * ImgSize]byte
	Label byte
}

Input is the type of input image

func (i *Input) Feature(n int) uint32

Feature extracts the n-th feature from input

func (i *Input) Output() uint16

func (i *Input) Parity() uint16

type SmallInput struct {
	Image [SmallImgSize * SmallImgSize]byte
	Label byte
}

SmallInput is the type of small input image

func (i *SmallInput) Feature(n int) uint32

Feature extracts the n-th feature from small input

func (i *SmallInput) Output() uint16

func (i *SmallInput) Parity() uint16