cdipaolo/goml

TFIDF doesn't work

Opened this issue · 1 comments

TFIDF doesn't work unless we actually save the DocsSeen value in the Bayes model.

Currently the struct for Word doesn't do this.

type Word struct { Count []uint64 Seen uint64 DocsSeen uint64 json:"-" }

Should be:

type Word struct { Count []uint64 Seen uint64 DocsSeen uint64 }