classifier

Documentation

Index

• Variables
• type BayesClassifier
  • func (classifier *BayesClassifier) BayesProbabilityOf(category string, features ...string) float64
  • func (classifier *BayesClassifier) Classify(features ...string) string
  • func (classifier *BayesClassifier) MolecularProbabilityOf(features ...string) []Classification
  • func (classifier *BayesClassifier) ProbabilityOf(features ...string) []Classification
• type Classification

Variables

var K = 1e3

var VERSION = "1.0.0"

Types

type BayesClassifier

type BayesClassifier struct {
	Brain *brain.BayesBrain
}

func (*BayesClassifier) BayesProbabilityOf

func (classifier *BayesClassifier) BayesProbabilityOf(category string, features ...string) float64

P(category|features...) 朴素贝叶斯计算特征数据属于某个类别的概率

func (*BayesClassifier) Classify

func (classifier *BayesClassifier) Classify(features ...string) string

func (*BayesClassifier) MolecularProbabilityOf

func (classifier *BayesClassifier) MolecularProbabilityOf(features ...string) []Classification

func (*BayesClassifier) ProbabilityOf

func (classifier *BayesClassifier) ProbabilityOf(features ...string) []Classification

type Classification

type Classification struct {
	Category    string
	Features    []string
	Probability float64
}