A Naive Bayes classifier in Golang
package main
import (
"fmt"
"github.com/ajph/nbclassifier-go"
)
func main() {
m := nbclassifier.New()
m.NewClass("banana")
m.Learn("banana", "sweet", "yellow", "long")
m.NewClass("orange")
m.Learn("orange", "sweet", "orange", "round")
m.NewClass("apple")
m.Learn("apple", "sweet", "green", "red", "round")
//m.SaveToFile("./test.json")
/*
// or load from existing file
m, err := nbclassifier.LoadFromFile("./test.json")
if err != nil {
panic(err)
}
*/
w, unsure, _ := m.Classify("round", "sweet")
fmt.Println(w.Id, unsure)
}- Documentation
- Tests