config

Documentation

Index

• Variables
• type Config
  • func (c Config) String() string

Variables

var (
	DefaultInputFile      string               = "example/input.txt"
	DefaultOutputFile     string               = "example/word_vectors.txt"
	DefaultDimension      int                  = 10
	DefaultIteration      int                  = 15
	DefaultMinCount       int                  = 5
	DefaultThreadSize     int                  = runtime.NumCPU()
	DefaultBatchSize      int                  = 10000
	DefaultWindow         int                  = 5
	DefaultInitlr         float64              = 0.025
	DefaultProf           bool                 = false
	DefaultToLower        bool                 = false
	DefaultVerbose        bool                 = false
	DefaultSaveVectorType model.SaveVectorType = model.NORMAL
	// Word2Vec
	DefaultModel              word2vec.ModelType     = word2vec.CBOW
	DefaultOptimizer          word2vec.OptimizerType = word2vec.NEGATIVE_SAMPLING
	DefaultMaxDepth           int                    = 0
	DefaultNegativeSampleSize int                    = 5
	DefaultSubsampleThreshold float64                = 1.0e-3
	DefaultTheta              float64                = 1.0e-4
	// GloVe
	DefaultSolver glove.SolverType = glove.SGD
	DefaultXmax   int              = 100
	DefaultAlpha  float64          = 0.75
	// Lexvex
	DefaultRelationType corpus.RelationType = corpus.PPMI
	DefaultSmooth       float64             = 0.75
	// Search
	DefaultRank int = 10
)

The defaults of Config.

Types

type Config

type Config int

Config is enum of the common config.

const (
	InputFile Config = iota
	OutputFile
	Dimension
	Iteration
	MinCount
	ThreadSize
	BatchSize
	Window
	Initlr
	Prof
	ToLower
	Verbose
	SaveVectorType
	// Word2Vec
	Model
	Optimizer
	MaxDepth
	NegativeSampleSize
	SubsampleThreshold
	Theta
	// GloVe
	Solver
	Xmax
	Alpha
	// Lexvec
	RelationType
	Smooth
	// Search
	Rank
)

The list of Config.

func (Config) String

func (c Config) String() string