Classifier

Cadmium comes with two classifiers so far, a Classic Bayes classifier and a Viterbi classifier.

Those are probabalistic classifiers that, when trained with a data set, can classify words (or other tokens) according to categories.

Installation

1. Add the dependency to your shard.yml:

   dependencies:
     cadmium_classifier:
       github: cadmiumcr/classifier

2. Run shards install

Usage

require "cadmium_classifier"

classifier = Cadmium::Classifier::Bayes.new

classifier.train("crystal is an awesome programming language", "programming")
classifier.train("ruby is nice, but not as fast as crystal", "programming")

classifier.train("my wife and I went to the beach", "off-topic")
classifier.train("my dog likes to go outside and play", "off-topic")

classifier.classify("Crystal is my favorite!")
# => "programming"

You can save the classifier as JSON as well

require "json"
json = classifier.to_json
File.write("classifier.json", json)

And load it again later

require "json"
json = File.open("classifier.json")
classifier = classifier.from_json(json)

Contributing

1. Fork it (https://github.com/cadmiumcr/classifier/fork)
2. Create your feature branch (git checkout -b my-new-feature)
3. Commit your changes (git commit -am 'Add some feature')
4. Push to the branch (git push origin my-new-feature)
5. Create a new Pull Request

Contributors

• Chris Watson - creator and maintainer