lexvec

Documentation

Index

• func LoadForCmd(cmd *cobra.Command, opts *Options)
• func New(opts ...ModelOption) (model.Model, error)
• func NewForOptions(opts Options) (model.Model, error)
• type ModelOption
  • func ToLower() ModelOption
  • func Verbose() ModelOption
  • func WithBatchSize(v int) ModelOption
  • func WithDimension(v int) ModelOption
  • func WithInitLearningRate(v float64) ModelOption
  • func WithIteration(v int) ModelOption
  • func WithMinCount(v int) ModelOption
  • func WithNegativeSampleSize(v int) ModelOption
  • func WithRelation(typ RelationType) ModelOption
  • func WithSmooth(v float64) ModelOption
  • func WithSubsampleThreshold(v float64) ModelOption
  • func WithTheta(v float64) ModelOption
  • func WithThreadSize(v int) ModelOption
  • func WithWindow(v int) ModelOption
• type Options
• type RelationType
  • func (t *RelationType) Set(name string) error
  • func (t *RelationType) String() string
  • func (t *RelationType) Type() string

Functions

func LoadForCmd

func LoadForCmd(cmd *cobra.Command, opts *Options)

func New

func New(opts ...ModelOption) (model.Model, error)

func NewForOptions

func NewForOptions(opts Options) (model.Model, error)

Types

type ModelOption

type ModelOption func(*Options)

func ToLower

func ToLower() ModelOption

func Verbose

func Verbose() ModelOption

func WithBatchSize

func WithBatchSize(v int) ModelOption

model options

func WithDimension

func WithDimension(v int) ModelOption

func WithInitLearningRate

func WithInitLearningRate(v float64) ModelOption

func WithIteration

func WithIteration(v int) ModelOption

func WithMinCount

func WithMinCount(v int) ModelOption

corpus options

func WithNegativeSampleSize

func WithNegativeSampleSize(v int) ModelOption

for lexvec options

func WithRelation

func WithRelation(typ RelationType) ModelOption

func WithSmooth

func WithSmooth(v float64) ModelOption

func WithSubsampleThreshold

func WithSubsampleThreshold(v float64) ModelOption

func WithTheta

func WithTheta(v float64) ModelOption

func WithThreadSize

func WithThreadSize(v int) ModelOption

func WithWindow

func WithWindow(v int) ModelOption

type Options

type Options struct {
	CorpusOptions corpus.Options
	ModelOptions  model.Options

	NegativeSampleSize int
	RelationType       RelationType
	Smooth             float64
	SubsampleThreshold float64
	Theta              float64
}

type RelationType

type RelationType string

const (
	PPMI           RelationType = "ppmi"
	PMI            RelationType = "pmi"
	Collocation    RelationType = "co"
	LogCollocation RelationType = "logco"
)

func (*RelationType) Set

func (t *RelationType) Set(name string) error

func (*RelationType) String

func (t *RelationType) String() string

func (*RelationType) Type

func (t *RelationType) Type() string