tfidf

Documentation

Index

• type Vectorizer
  • func NewVectorizer() *Vectorizer
  • func (v *Vectorizer) Fit(docs []string)
  • func (v *Vectorizer) FitTransform(docs []string) [][]float64
  • func (v *Vectorizer) Transform(docs []string) [][]float64

Types

type Vectorizer

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

Vectorizer represents a TF-IDF vectorizer

func NewVectorizer

func NewVectorizer() *Vectorizer

NewVectorizer creates a new Vectorizer

func (*Vectorizer) Fit

func (v *Vectorizer) Fit(docs []string)

Fit fits the vectorizer to the input documents

func (*Vectorizer) FitTransform

func (v *Vectorizer) FitTransform(docs []string) [][]float64

FitTransform fits the vectorizer to the input documents and then transforms them

func (*Vectorizer) Transform

func (v *Vectorizer) Transform(docs []string) [][]float64

Transform transforms the input documents to TF-IDF vectors