bayes

Documentation

Index

• type Model
  • func NewModel(account string) *Model
  • func (m *Model) Infer(t *syntax.Transaction)
  • func (m *Model) Update(t *syntax.Transaction)

Types

type Model

type Model struct {
	// contains filtered or unexported fields
}

Model implements a Bayes model for accounts and text tokens derived from transactions.

func NewModel

func NewModel(account string) *Model

NewModel creates a new model.

func (*Model) Infer

func (m *Model) Infer(t *syntax.Transaction)

Infer replaces the given account with an inferred account. P(A | T1 & T2 & ... & Tn) ~ P(A) * P(T1|A) * P(T2|A) * ... * P(Tn|A)

func (*Model) Update

func (m *Model) Update(t *syntax.Transaction)

Update updates the model with the given transaction.