training

Documentation

Index

• func NewMNISTReader(dataFilename string, resultsFilename string, validatorFilename string, ...) (r *mnistReader)
• type TextDataReader
  • func NewTextDataReader(filename string, validationPart int) *TextDataReader
  • func (r *TextDataReader) DataCount() int
  • func (r *TextDataReader) GetData(i int) (*mat.Dense, *mat.Dense)
  • func (r *TextDataReader) GetValidator(i int) (*mat.Dense, *mat.Dense)
  • func (r *TextDataReader) ValidatorCount() int
• type Trainer

Functions

func NewMNISTReader

func NewMNISTReader(dataFilename string, resultsFilename string, validatorFilename string, validatorResultsFilename string) (r *mnistReader)

Types

type TextDataReader

type TextDataReader struct {
	// contains filtered or unexported fields
}

func NewTextDataReader

func NewTextDataReader(filename string, validationPart int) *TextDataReader

func (*TextDataReader) DataCount

func (r *TextDataReader) DataCount() int

func (*TextDataReader) GetData

func (r *TextDataReader) GetData(i int) (*mat.Dense, *mat.Dense)

func (*TextDataReader) GetValidator

func (r *TextDataReader) GetValidator(i int) (*mat.Dense, *mat.Dense)

func (*TextDataReader) ValidatorCount

func (r *TextDataReader) ValidatorCount() int

type Trainer

type Trainer interface {
	GetData(i int) (*mat.Dense, *mat.Dense)
	DataCount() int

	GetValidator(i int) (*mat.Dense, *mat.Dense)
	ValidatorCount() int
}

Trainer is basic inteface for neuralnetwork.NeuralNetwork training and validation