brain

package
v1.4.5 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Aug 12, 2020 License: Apache-2.0 Imports: 6 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source
var VERSION = "v0.0.3"

Functions

This section is empty.

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 added in v1.2.4

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 added in v1.2.4

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

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL