Skip to content

cadmiumcr/classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published