123456789101112131415161718192021222324252627282930313233 |
- 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)
- }
- }
|