rubiks

Documentation

Index

• type Prediction
  • func NewPrediction(input string, runeToIdx func(r rune) (int, error), sampleSize, vocabSize int) *Prediction
  • func (p *Prediction) GetOutput() [][]float32
  • func (p *Prediction) Read() ([]float32, error)
  • func (p *Prediction) Write(val []float32) error
• type TrainingSet
  • func NewTrainingSet() *TrainingSet
  • func (t *TrainingSet) GetTrainer() (datasetter.Trainer, error)

Types

type Prediction

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

Prediction is the based type that can be used as a training dataset

func NewPrediction

func NewPrediction(input string, runeToIdx func(r rune) (int, error), sampleSize, vocabSize int) *Prediction

NewPrediction return an object suitable for the LSTM

func (*Prediction) GetOutput

func (p *Prediction) GetOutput() [][]float32

GetOutput ...

func (*Prediction) Read

func (p *Prediction) Read() ([]float32, error)

Float32Read ...

func (*Prediction) Write

func (p *Prediction) Write(val []float32) error

Float32Write ...

type TrainingSet

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

func NewTrainingSet

func NewTrainingSet() *TrainingSet

NewTrainingSet from a ReadSeeker

func (*TrainingSet) GetTrainer

func (t *TrainingSet) GetTrainer() (datasetter.Trainer, error)