omnicat-bayes 0.2.1 → 0.2.2

data/lib/omnicat/bayes/version.rb

@@ -1,5 +1,5 @@
 module Omnicat
   module Bayes
-    VERSION = '0.2.1'
+    VERSION = '0.2.2'
   end
 end

data/lib/omnicat/classifiers/bayes.rb

@@ -96,8 +96,9 @@ module OmniCat
                   "Document is not found in #{category_name} documents!"
           end
           doc.tokens.each do |token, count|
-            decrement_token_counts(category_name, token, count)
             @categories[category_name].tokens[token] = @categories[category_name].tokens[token].to_i - count
+            @categories[category_name].tokens.delete(token) if @categories[category_name].tokens[token] == 0
+            decrement_token_counts(category_name, token, count)
           end
           @categories[category_name].docs.delete(doc_key) if @categories[category_name].docs[doc_key].count == 0
           decrement_doc_counts(category_name)
@@ -180,7 +181,7 @@ module OmniCat
 
         # nodoc
         def modify_uniq_token_count(token, uniq_token_addition)
-          categories.each do |_, category|
+          @categories.each do |_, category|
              if category.tokens.has_key?(token)
                uniq_token_addition = 0
                break

data/omnicat-bayes.gemspec

@@ -18,7 +18,8 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ['lib']
   
-  spec.add_dependency 'omnicat', '~> 0.2.1'
+  spec.add_dependency 'hashable', '~> 0.1.1'
+  spec.add_dependency 'omnicat', '~> 0.2.2'
   spec.add_development_dependency 'bundler', '~> 1.3'
   spec.add_development_dependency 'rake'
 end

data/test/unit/classifiers/bayes_test.rb

@@ -128,6 +128,16 @@ class TestBayes < Test::Unit::TestCase
   def test_train_missing_category
     assert_raise(StandardError) { @bayes.train 'neutral', 'how are you?' }
   end
+  
+  def test_unique_token_count
+    @bayes.add_category 'positive'
+    @bayes.train_batch 'positive', ['good job ever', 'valid syntax',
+      'best moments of my good life']
+    assert_equal(10,@bayes.uniq_token_count)
+    @bayes.untrain_batch 'positive', ['good job ever', 'valid syntax',
+      'best moments of my good life']
+    assert_equal(0,@bayes.uniq_token_count)
+  end
 
   def test_classifiability_error
     @bayes.add_category 'positive'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omnicat-bayes
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
   prerelease: 
 platform: ruby
 authors:
@@ -11,6 +11,22 @@ bindir: bin
 cert_chain: []
 date: 2013-07-11 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: hashable
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.1.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.1.1
 - !ruby/object:Gem::Dependency
   name: omnicat
   requirement: !ruby/object:Gem::Requirement
@@ -18,7 +34,7 @@ dependencies:
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: 0.2.1
+        version: 0.2.2
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -26,7 +42,7 @@ dependencies:
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: 0.2.1
+        version: 0.2.2
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement