Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

https://github.com/eaigner/shield

Bayesian text classifier with flexible tokenizers and storage backends for Go
https://github.com/eaigner/shield

Last synced: about 2 months ago
JSON representation

Bayesian text classifier with flexible tokenizers and storage backends for Go

Lists

README 

        
Shield is a bayesian text classifier with flexible tokenizer and backend store support



Currently implemented:



- Redis backend

- English tokenizer



## Example



```go

package main



import (

  "github.com/eaigner/shield"

)



func main() {

  sh := shield.New(

    shield.NewEnglishTokenizer(),

    shield.NewRedisStore("127.0.0.1:6379", "", 0),

  )



  sh.Learn("good", "sunshine drugs love sex lobster sloth")

  sh.Learn("bad", "fear death horror government zombie god")



  c, _ := sh.Classify("sloths are so cute i love them")

  if c != "good" {

    panic(c)

  }



  c, _ = sh.Classify("i fear god and love the government")

  if c != "bad" {

    panic(c)

  }

}

```