builder

package
v0.0.0-...-ba2758a Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Nov 20, 2019 License: Apache-2.0 Imports: 8 Imported by: 1

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type GloveBuilder 

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

GloveBuilder manages the members to build Model interface.

func NewGloveBuilder 

func NewGloveBuilder() *GloveBuilder

NewGloveBuilder creates *GloveBuilder

func NewGloveBuilderFromViper 

func NewGloveBuilderFromViper() (*GloveBuilder, error)

NewGloveBuilderFromViper creates *GloveBuilder from viper.

func (*GloveBuilder) Alpha 

func (gb *GloveBuilder) Alpha(alpha float64) *GloveBuilder

Alpha sets alpha.

func (*GloveBuilder) BatchSize 

func (gb *GloveBuilder) BatchSize(batchSize int) *GloveBuilder

BatchSize sets batch size to to preprocess/train.

func (*GloveBuilder) Build 

func (gb *GloveBuilder) Build() (model.Model, error)

Build creates model.Model interface.

func (*GloveBuilder) Dimension 

func (gb *GloveBuilder) Dimension(dimension int) *GloveBuilder

Dimension sets dimension of word vector.

func (*GloveBuilder) Initlr 

func (gb *GloveBuilder) Initlr(initlr float64) *GloveBuilder

Initlr sets initial learning rate.

func (*GloveBuilder) Iteration 

func (gb *GloveBuilder) Iteration(iter int) *GloveBuilder

Iteration sets number of iteration.

func (*GloveBuilder) MinCount 

func (gb *GloveBuilder) MinCount(minCount int) *GloveBuilder

MinCount sets min count.

func (*GloveBuilder) SaveVectorType 

func (gb *GloveBuilder) SaveVectorType(typ model.SaveVectorType) *GloveBuilder

func (*GloveBuilder) Solver 

func (gb *GloveBuilder) Solver(typ glove.SolverType) *GloveBuilder

Solver sets solver.

func (*GloveBuilder) ThreadSize 

func (gb *GloveBuilder) ThreadSize(threadSize int) *GloveBuilder

ThreadSize sets number of goroutine.

func (*GloveBuilder) ToLower 

func (gb *GloveBuilder) ToLower() *GloveBuilder

ToLower is whether converts the words in corpus to lowercase or not.

func (*GloveBuilder) Verbose 

func (gb *GloveBuilder) Verbose() *GloveBuilder

Verbose sets verbose mode.

func (*GloveBuilder) Window 

func (gb *GloveBuilder) Window(window int) *GloveBuilder

Window sets context window size.

func (*GloveBuilder) Xmax 

func (gb *GloveBuilder) Xmax(xmax int) *GloveBuilder

Xmax sets x-max.

type LexvecBuilder 

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

LexvecBuilder manages the members to build Model interface.

func NewLexvecBuilder 

func NewLexvecBuilder() *LexvecBuilder

NewLexvecBuilder creates *LexvecBuilder.

func NewLexvecBuilderFromViper 

func NewLexvecBuilderFromViper() (*LexvecBuilder, error)

NewLexvecBuilderFromViper creates *LexvecBuilder from viper.

func (*LexvecBuilder) BatchSize 

func (lb *LexvecBuilder) BatchSize(batchSize int) *LexvecBuilder

BatchSize sets batch size to preprocess/train.

func (*LexvecBuilder) Build 

func (lb *LexvecBuilder) Build() (model.Model, error)

Build creates Lexvec model.

func (*LexvecBuilder) Dimension 

func (lb *LexvecBuilder) Dimension(dimension int) *LexvecBuilder

Dimension sets dimension of word vector.

func (*LexvecBuilder) Initlr 

func (lb *LexvecBuilder) Initlr(initlr float64) *LexvecBuilder

Initlr sets initial learning rate.

func (*LexvecBuilder) Iteration 

func (lb *LexvecBuilder) Iteration(iter int) *LexvecBuilder

Iteration sets number of iteration.

func (*LexvecBuilder) MinCount 

func (lb *LexvecBuilder) MinCount(minCount int) *LexvecBuilder

MinCount sets min count.

func (*LexvecBuilder) NegativeSampleSize 

func (lb *LexvecBuilder) NegativeSampleSize(size int) *LexvecBuilder

NegativeSampleSize sets number of samples as negative.

func (*LexvecBuilder) RelationType 

func (lb *LexvecBuilder) RelationType(typ corpus.RelationType) *LexvecBuilder

func (*LexvecBuilder) SaveVectorType 

func (lb *LexvecBuilder) SaveVectorType(typ model.SaveVectorType) *LexvecBuilder

func (*LexvecBuilder) Smooth 

func (lb *LexvecBuilder) Smooth(smooth float64) *LexvecBuilder

func (*LexvecBuilder) SubSampleThreshold 

func (lb *LexvecBuilder) SubSampleThreshold(threshold float64) *LexvecBuilder

SubSampleThreshold sets threshold for subsampling.

func (*LexvecBuilder) Theta 

func (lb *LexvecBuilder) Theta(theta float64) *LexvecBuilder

func (*LexvecBuilder) ThreadSize 

func (lb *LexvecBuilder) ThreadSize(threadSize int) *LexvecBuilder

ThreadSize sets number of goroutine.

func (*LexvecBuilder) ToLower 

func (lb *LexvecBuilder) ToLower() *LexvecBuilder

ToLower is whether converts the words in corpus to lowercase or not.

func (*LexvecBuilder) Verbose 

func (lb *LexvecBuilder) Verbose() *LexvecBuilder

Verbose sets verbose mode.

func (*LexvecBuilder) Window 

func (lb *LexvecBuilder) Window(window int) *LexvecBuilder

Window sets context window size.

type Word2vecBuilder 

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

Word2vecBuilder manages the members to build Model interface.

func NewWord2vecBuilder 

func NewWord2vecBuilder() *Word2vecBuilder

NewWord2vecBuilder creates *Word2vecBuilder.

func NewWord2vecBuilderFromViper 

func NewWord2vecBuilderFromViper() (*Word2vecBuilder, error)

NewWord2vecBuilderFromViper creates *Word2vecBuilder from viper.

func (*Word2vecBuilder) BatchSize 

func (wb *Word2vecBuilder) BatchSize(batchSize int) *Word2vecBuilder

BatchSize sets batch size to to preprocess/train.

func (*Word2vecBuilder) Build 

func (wb *Word2vecBuilder) Build() (model.Model, error)

Build creates model.Model interface.

func (*Word2vecBuilder) Dimension 

func (wb *Word2vecBuilder) Dimension(dimension int) *Word2vecBuilder

Dimension sets dimension of word vector.

func (*Word2vecBuilder) Initlr 

func (wb *Word2vecBuilder) Initlr(initlr float64) *Word2vecBuilder

Initlr sets initial learning rate.

func (*Word2vecBuilder) Iteration 

func (wb *Word2vecBuilder) Iteration(iter int) *Word2vecBuilder

Iteration sets number of iteration.

func (*Word2vecBuilder) MaxDepth 

func (wb *Word2vecBuilder) MaxDepth(maxDepth int) *Word2vecBuilder

MaxDepth sets number of times to track huffman tree.

func (*Word2vecBuilder) MinCount 

func (wb *Word2vecBuilder) MinCount(minCount int) *Word2vecBuilder

MinCount sets min count.

func (*Word2vecBuilder) Model 

func (wb *Word2vecBuilder) Model(typ word2vec.ModelType) *Word2vecBuilder

Model sets model of Word2vec. One of: cbow|skip-gram

func (*Word2vecBuilder) NegativeSampleSize 

func (wb *Word2vecBuilder) NegativeSampleSize(size int) *Word2vecBuilder

NegativeSampleSize sets number of samples as negative.

func (*Word2vecBuilder) Optimizer 

func (wb *Word2vecBuilder) Optimizer(typ word2vec.OptimizerType) *Word2vecBuilder

Optimizer sets optimizer of Word2vec. One of: hs|ns

func (*Word2vecBuilder) SaveVectorType 

func (wb *Word2vecBuilder) SaveVectorType(typ model.SaveVectorType) *Word2vecBuilder

func (*Word2vecBuilder) SubSampleThreshold 

func (wb *Word2vecBuilder) SubSampleThreshold(threshold float64) *Word2vecBuilder

SubSampleThreshold sets threshold for subsampling.

func (*Word2vecBuilder) Theta 

func (wb *Word2vecBuilder) Theta(theta float64) *Word2vecBuilder

Theta sets lower limit of learning rate (lr >= initlr * theta).

func (*Word2vecBuilder) ThreadSize 

func (wb *Word2vecBuilder) ThreadSize(threadSize int) *Word2vecBuilder

ThreadSize sets number of goroutine.

func (*Word2vecBuilder) ToLower 

func (wb *Word2vecBuilder) ToLower() *Word2vecBuilder

ToLower is whether converts the words in corpus to lowercase or not.

func (*Word2vecBuilder) Verbose 

func (wb *Word2vecBuilder) Verbose() *Word2vecBuilder

Verbose sets verbose mode.

func (*Word2vecBuilder) Window 

func (wb *Word2vecBuilder) Window(window int) *Word2vecBuilder

Window sets context window size.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.