SimpleNaiveBayes

This is a very simple naive bayes written in ruby.

Installation

$ gem install simple_naive_bayes

Usage

require 'simple_naive_bayes'
cl = SimpleNaiveBayes::NaiveBayes.new
cl.training("yes", ["Chinese", "Beijing", "Chinese"])
cl.training("yes", ["Chinese", "Chinese", "Shanghai"])
cl.training("yes", ["Chinese", "Macao"])
cl.training("no", ["Tokyo", "Japan", "Chinese"])

cl.classify(["Tokyo"])

show example.rb

Supported Ruby Versions

Ruby 2.0.0

Performance

To measure the performance of the filte, I tested.
The datasource is publiccorpus (http://spamassassin.apache.org/publiccorpus/).
This data is mail corpus, so I classify mails.
Those mails have three type which is spam, easy_ham, hard_ham.
The test script is publiccorpus_test.rb.

Data sources

Training Data

Test Data

Result

spam accuracy rate is 99.6% (498/500)
easy ham accuracy rate is 99.8% (2497/2500)
hard ham accuracy rate is 81.6% (204/250)

License

MIT License

Contributing

Fork it
Create your feature branch (git checkout -b my-new-feature)
Commit your changes (git commit -am 'Add some feature')
Push to the branch (git push origin my-new-feature)
Create new Pull Request