Shield is a bayesian text classifier with flexible tokenizer and backend store support
Currently implemented:
- Redis backend
- English tokenizer
package main
import (
"github.com/eaigner/shield"
)
funcmain() {
sh:=shield.New(
shield.NewEnglishTokenizer(),
shield.NewRedisStore("127.0.0.1:6379", "", 0),
)
sh.Learn("good", "sunshine drugs love sex lobster sloth")
sh.Learn("bad", "fear death horror government zombie god")
c, _:=sh.Classify("sloths are so cute i love them")
ifc!="good" {
panic(c)
}
c, _=sh.Classify("i fear god and love the government")
ifc!="bad" {
panic(c)
}
}