glove

type AdaGrad ¶

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

AdaGrad behaviors as one of Glove solver.

func NewAdaGrad(dimension int, initlr float64) *AdaGrad

type Glove ¶

type Glove struct {
	*model.Option
	*GloveOption
	*corpus.CountModelCorpus
	// contains filtered or unexported fields
}

Glove stores the configs for Glove models.

func NewGlove(option *model.Option, gloveOption *GloveOption) *Glove

func (g *Glove) Get() (map[string][]float64, error)

func (g *Glove) Save(outputPath string) error

func (g *Glove) Train(f io.Reader) error

type GloveOption ¶

type GloveOption struct {
	Solver Solver
	Xmax   int
	Alpha  float64
}

type Sgd ¶

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

Sgd is stochastic gradient descent that behaviors as one of GloVe solver.

func NewSgd(dimension int, initlr float64) *Sgd

type Solver ¶

type Solver interface {
	// contains filtered or unexported methods
}

Solver is the interface for training with GloVe.

type SolverType ¶

type SolverType int

const (
	SGD SolverType = iota
	ADAGRAD
)

func (t SolverType) String() string