func main() { util.SeedNow() corpus := load.Newsgroups.Import() lda := topic.NewLDA(corpus, 20, .1, .01) graphical.RunGibbs(lda, 100) labels := eval.NewLabelCorpusWordFeature(corpus, lda.Z) fmt.Println("Accuracty:", eval.Naive(labels.SplitRand(.8))) for z := 0; z < lda.T; z++ { fmt.Printf("%d: %s\n", z, lda.TopicSummary(z, 10)) } }
func MMGibbs(c *pipeline.Corpus) []int { mom := MOM(c) graphical.RunGibbs(mom, 30) return mom.Z }
func DPGibbs(c *pipeline.Corpus) []int { dpmom := DPMOM(c) graphical.RunGibbs(dpmom, 30) return dpmom.Z }