github.com/qiuhoude/go-web@v0.0.0-20220223060959-ab545e78f20d/prepare/15bayesian/test/main.go (about)

     1  package main
     2  
     3  import (
     4  	"fmt"
     5  	"github.com/jbrukh/bayesian"
     6  )
     7  
     8  const (
     9  	Good bayesian.Class = "Good"
    10  	Bad  bayesian.Class = "Bad"
    11  )
    12  
    13  func main() {
    14  	classifier := bayesian.NewClassifier(Good, Bad)
    15  	goodStuff := []string{"tall", "rich", "handsome"}
    16  	badStuff := []string{"poor", "smelly", "ugly"}
    17  	classifier.Learn(goodStuff, Good)
    18  	classifier.Learn(badStuff, Bad)
    19  
    20  	classifier.ConvertTermsFreqToTfIdf()
    21  
    22  	scores, likely, _ := classifier.LogScores([]string{"tall", "girl"})
    23  
    24  	fmt.Println(scores, likely)
    25  }