nb 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e163cd57a340190efef6d152493f3cbf5f0ee6c7
-  data.tar.gz: 3dc32f2eaf69f8cda5fe9672bdd54c4c51d47a57
+  metadata.gz: 0d55c868cf9d3df0866130593f9e7d78935d9fde
+  data.tar.gz: 5e9cd9507bdea2a306e3a8fa9423153d0d4cfbb5
 SHA512:
-  metadata.gz: fb8c29aa398fdda618c7c624146435a5d38386f4698f4a4262c5178826089e50568b5eb213af53f7f1de48d5e73f93e5fad9cbf47868403f4a522350f03c1b73
-  data.tar.gz: 3e3bc9a7f6371ad037fb62a1cf3a5cbeb747aa958bea96c8a14879174a3bfd754a7580612961916f6be926b9cc56a80be28add65572e8ad376ce274caf1a1fa5
+  metadata.gz: 299259d6e2322cfe07adca36f84899796632bbc7f36c407de3a6e28d763355ba7518cabc82c7f3e8339c8783fa9937cb14edfc062d50946832c0d112c72d4c44
+  data.tar.gz: 243a9a559861ea695f66f21fe6b34bd5ecc89809b819306c1a9833f08bf51cad1131846180fa6dc55aaad404dc97a13b708ce281b9f95c499691da09e65a6e75

data/README.md

@@ -2,6 +2,7 @@
 
 [![Code Climate](https://codeclimate.com/github/forresty/nb/badges/gpa.svg)](https://codeclimate.com/github/forresty/nb)
 [![Build Status](https://travis-ci.org/forresty/nb.svg?branch=master)](https://travis-ci.org/forresty/nb)
+[![Gem Version](https://badge.fury.io/rb/nb.svg)](http://badge.fury.io/rb/nb)
 
 yet another Naive Bayes library
 
@@ -60,9 +61,15 @@ bayes.classify(*%w{ love }).should == [:love, 0.5]
 +------------+------+--------------------+
 ```
 
+## Credits
+
+- [classifier gem](https://github.com/cardmagic/classifier)
+- [naive_bayes gem](https://github.com/reddavis/Naive-Bayes)
+- [nbayes gem](https://github.com/oasic/nbayes)
+
 ## Contributing
 
-1. Fork it ( https://github.com/[my-github-username]/nb/fork )
+1. Fork it ( https://github.com/forresty/nb/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`)

data/lib/nb/naive_bayes.rb

@@ -1,12 +1,13 @@
 require "yaml"
 
 class NaiveBayes
-  attr_accessor :categories, :tokens_count, :categories_count
+  attr_accessor :categories, :tokens_count, :categories_count, :default_category
 
   def initialize(*categories)
     @categories = categories
     @tokens_count = {}
     @categories_count = {}
+    @default_category = @categories.first
 
     categories.each do |category|
       @tokens_count[category] = Hash.new(0)
@@ -21,8 +22,21 @@ class NaiveBayes
     @categories_count[category] += 1
   end
 
+  def untrain(category, *tokens)
+    tokens.uniq.each do |token|
+      @tokens_count[category][token] -= 1
+    end
+    @categories_count[category] -= 1
+  end
+
   def classify(*tokens)
-    classifications(*tokens).first
+    result = classifications(*tokens).first
+
+    if result.last == 0.0
+      [@default_category, 0.0]
+    else
+      result
+    end
   end
 
   def classifications(*tokens)

data/lib/nb/version.rb

@@ -1,3 +1,3 @@
 class NaiveBayes
-  VERSION = "0.0.2"
+  VERSION = "0.0.4"
 end

data/spec/nb/naive_bayes_spec.rb

@@ -2,6 +2,7 @@ require "spec_helper"
 
 describe NaiveBayes do
   it { should respond_to :train }
+  it { should respond_to :untrain }
   it { should respond_to :save }
   it { should respond_to :classify }
   it { should respond_to :classifications }
@@ -12,6 +13,7 @@ describe NaiveBayes do
   # it { should respond_to :total_number_of_tokens }
   it { should respond_to :total_number_of_items }
   it { should respond_to :top_tokens_of_category }
+  it { should respond_to :default_category= }
 
   let(:bayes) { NaiveBayes.new(:love, :hate) }
   subject { bayes }

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nb
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Forrest Ye
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-18 00:00:00.000000000 Z
+date: 2014-11-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler