simple_naive_bayes 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fe3a97c568b49df1ee13a85badfa0c106e800f84
-  data.tar.gz: 3ce26714bcbbd3cd914dd96b996447e6c0150db4
+  metadata.gz: a7d656f37e20a851ca3011fd417a53c66c8c0804
+  data.tar.gz: 9d62b66d9634bdde996456a734addffdf1bcc417
 SHA512:
-  metadata.gz: 652a9f57aa077f89e7d3611649630684bf4a49e6236e886685952d433d2081ba65a2d7c78fa8d1d0085f458378e30ed46a6e3e64e82659af10e78c39b8433151
-  data.tar.gz: d313812706be35cedaa51cbe93b3db5d41fbb6eb1659d733ec1b4556a7732ac14777bc56c731b269260537112a4a4aaf4746738bc96cd82bc51fd4475a5712d4
+  metadata.gz: 8d8c8f8940bc942bc0811d2b790776a656ecd42f5fc060189642f36123c79c8d14df2e6566601a82313de1245e0b39b542339d812bcacc7c0fbfcc49fc1f1713
+  data.tar.gz: 73c7c940206a57020ca1e34e1d843596345102e8d72663bb42e2d3175ef236b9a6047fce43c1de906ae9618a1a5136ab07ad944c2eedb3e5d81fb6fcbd4b7970

data/lib/simple_naive_bayes/version.rb

@@ -1,3 +1,3 @@
 module SimpleNaiveBayes
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/simple_naive_bayes.rb

@@ -68,6 +68,8 @@ module SimpleNaiveBayes
       def classify(doc)
         result = classify_with_all_result(doc)
 
+        return nil unless result
+
         best = result.max_by { |classify_relust| classify_relust[1] }
         best[0]
       end
@@ -77,13 +79,14 @@ module SimpleNaiveBayes
       # return [ [category1, probability1], [category2, probability2]... ]
       def classify_with_all_result(doc)
         result = []
+        return nil unless doc
 
         @categories_count.keys().each do |category|
           # log(P(doc|cat))
           document_category = calc_document_category(doc, category)
 
           # log(P(cat)) =  log(@categories_count[cat]) - log( @all_category_num )
-          category_probability = Math.log2(@categories_count[category]) - Math.log2(@all_category_num)
+          category_probability = calc_category_probability(category)
 
           # log(P(cat|doc)) = log(P(doc|cat)) + log(P(cat))
           category_document_probability = document_category + category_probability
@@ -93,6 +96,10 @@ module SimpleNaiveBayes
         result
       end
 
+      def calc_category_probability(category)
+        Math.log2(@categories_count[category]) - Math.log2(@all_category_num)
+      end
+
       # log(P(doc|cat)) = log(P(word1|cat)) + log(P(word2|cat)) + ....
       def calc_document_category(doc, category)
         probability = 0

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: simple_naive_bayes
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - y42sora
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-08-17 00:00:00.000000000 Z
+date: 2013-08-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler