brain

Documentation

Index

• Variables
• type BayesBrain
  • func NewBayesBrain() *BayesBrain
  • func (brain *BayesBrain) ApplyTfIdf()
  • func (brain *BayesBrain) Learn(category string, features ...string)
  • func (brain *BayesBrain) Load(filename string) error
  • func (brain *BayesBrain) Save(filename string) error
  • func (brain *BayesBrain) Show()
• type CategorySummary

Variables

var VERSION = "v0.0.3"

Types

type BayesBrain

type BayesBrain struct {
	FeaturesFrequency               map[string]float64
	CategoriesFrequency             map[string]float64
	FeaturesFrequencyInEachCategory map[string]map[string]float64
	CategoriesSummary               map[string]*CategorySummary
	LearnedCount                    int
	DidConvertTfIdf                 bool

	TfIdfTempValues map[string]map[string]float64
}

func NewBayesBrain

func NewBayesBrain() *BayesBrain

func (*BayesBrain) ApplyTfIdf

func (brain *BayesBrain) ApplyTfIdf()

TF-IDF https://en.wikipedia.org/wiki/Tf%E2%80%93idf

func (*BayesBrain) Learn

func (brain *BayesBrain) Learn(category string, features ...string)

func (*BayesBrain) Load

func (brain *BayesBrain) Load(filename string) error

func (*BayesBrain) Save

func (brain *BayesBrain) Save(filename string) error

func (*BayesBrain) Show

func (brain *BayesBrain) Show()

type CategorySummary

type CategorySummary struct {
	Tfs          map[string][]float64
	LearnedCount int
}