simple_naive_bayes 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a7d656f37e20a851ca3011fd417a53c66c8c0804
-  data.tar.gz: 9d62b66d9634bdde996456a734addffdf1bcc417
+  metadata.gz: 2fb9d56e4001ff6ebff3b86a05bf0b77d21180f7
+  data.tar.gz: cf92585d78df2861eff5625e32e4f646d8f1c501
 SHA512:
-  metadata.gz: 8d8c8f8940bc942bc0811d2b790776a656ecd42f5fc060189642f36123c79c8d14df2e6566601a82313de1245e0b39b542339d812bcacc7c0fbfcc49fc1f1713
-  data.tar.gz: 73c7c940206a57020ca1e34e1d843596345102e8d72663bb42e2d3175ef236b9a6047fce43c1de906ae9618a1a5136ab07ad944c2eedb3e5d81fb6fcbd4b7970
+  metadata.gz: 09ba13da3fccea105957f657bc652c63884127485296f83f75d5ada050d9e85ed4333c81d141c0ab7c834a509a9600bbb570a15969ac9e84d7948d70969015e7
+  data.tar.gz: 2c4051dfbed05e50f5160516f739b0257aa1ccee7da27079bd0c5d634e462678630542d3a7af992d1f8e4b4b7dc6124abd187776c73d89ae6b1baa47bafc2890

data/example/example.rb

@@ -16,4 +16,15 @@ end
 test = ["Chinese", "Chinese", "Chinese", "Tokyo", "Japan"]
 
 p cl.classify(test)
-p cl.classify_with_all_result(test)
+p cl.classify_with_all_result(test)
+
+
+
+cl = SimpleNaiveBayes::NaiveBayes.new(1)
+
+data.each do |cat, doc|
+  cl.training(cat, doc)
+end
+
+p cl.classify(test)
+p cl.classify_with_all_result(test)

data/lib/simple_naive_bayes/version.rb

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

data/lib/simple_naive_bayes.rb

@@ -34,7 +34,7 @@ module SimpleNaiveBayes
     http://aidiary.hatenablog.com/entry/20100613/1276389337
 =end
 
-      def initialize()
+      def initialize(additive = 0.5)
         @all_category_num = 0
         @all_word_set = Set.new
         @categories_count = Hash.new(0)
@@ -42,7 +42,7 @@ module SimpleNaiveBayes
         @categories_word = Hash.new
         @categories_all_word_count = Hash.new(0)
         @laplace_categories_all_word_count = Hash.new(0)
-        @additive = 0.5
+        @additive = additive
       end
 
       """

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: simple_naive_bayes
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - y42sora