34 lines
536 B
Go
34 lines
536 B
Go
package multibayes
|
|
|
|
type document struct {
|
|
Text string
|
|
Classes []string
|
|
}
|
|
|
|
func getTestData() []document {
|
|
|
|
documents := []document{
|
|
{
|
|
Text: "My dog has fleas.",
|
|
Classes: []string{"vet"},
|
|
},
|
|
{
|
|
Text: "My cat has ebola.",
|
|
Classes: []string{"vet", "cdc"},
|
|
},
|
|
{
|
|
Text: "Aaron has ebola.",
|
|
Classes: []string{"cdc"},
|
|
},
|
|
}
|
|
|
|
return documents
|
|
}
|
|
|
|
func (c *Classifier) trainWithTestData() {
|
|
testdata := getTestData()
|
|
for _, document := range testdata {
|
|
c.Add(document.Text, document.Classes)
|
|
}
|
|
}
|