Versions in this module Expand all Collapse all v0 v0.0.47 Oct 31, 2023 v0.0.46 Oct 1, 2023 Changes in this version + type NaiveBayesClassifier struct + CategoriesDocuments map[string]int + CategoriesWords map[string]int + Threshold float64 + TotalDocuments int + TotalWords int + Words map[string]map[string]int + func New(threshold float64) *NaiveBayesClassifier + func NewClassifierFromFile(path string) (*NaiveBayesClassifier, error) + func NewClassifierFromFileData(data []byte) (*NaiveBayesClassifier, error) + func NewClassifierWithReader(reader io.Reader) (*NaiveBayesClassifier, error) + func (c *NaiveBayesClassifier) Classify(document string) (category string) + func (c *NaiveBayesClassifier) Evaluate(train, test []mlutils.LabeledDocument) + func (c *NaiveBayesClassifier) Fit(data map[string][]string) + func (c *NaiveBayesClassifier) Probabilities(document string) (p map[string]float64) + func (c *NaiveBayesClassifier) SaveClassifierToFile(path string) error + type Sorted struct + Category string + Probability float64