nbayes 0.1.0 → 0.1.1

data/README.rdoc

@@ -12,7 +12,7 @@ NBayes is a full-featured, Ruby implementation of Naive Bayes.  Some of the feat
 * optional binarized mode
 * uses log probabilities to avoid underflow
 
-For more information, view this blog post (coming soon)
+For more information, view this blog post: http://blog.oasic.net/2012/06/naive-bayes-for-ruby.html
 
 == Contributing to nbayes
  

data/VERSION

@@ -1 +1 @@
-0.1.0
+0.1.1

data/lib/nbayes.rb

@@ -159,7 +159,6 @@ module NBayes
       final_probs
     end
 
-
     # Loads class instance from a data file (e.g., yaml)
     def self.from(yml_file)
       nbayes = YAML.load_file(yml_file)
@@ -167,9 +166,26 @@ module NBayes
       nbayes
     end
 
-    # Dumps class instance to a file
-    def dump(yml_file)
-      File.open(yml_file, "w") {|f| YAML.dump(self, f) }
+    # Load class instance
+    def load(yml)
+      if yml.nil?
+	return NBayes::Base.new
+      elsif yml[0..2] == "---"
+        nbayes = YAML.load(yml)
+      else
+        nbayes = YAML.load_file(yml_file)
+      end
+      nbayes.reset_after_import()         		# yaml does not properly set the defaults on the Hashes
+      nbayes
+    end
+    
+    # Dumps class instance to a data file (e.g., yaml) or a string
+    def dump(arg)
+      if arg.instance_of? String
+        File.open(arg, "w") {|f| YAML.dump(self, f) }
+      else
+        YAML.dump(arg)
+      end
     end
 
   end

data/nbayes.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "nbayes"
-  s.version = "0.1.0"
+  s.version = "0.1.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["oasic"]
-  s.date = "2012-06-27"
+  s.date = "2012-07-13"
   s.description = "Ruby implementation of Naive Bayes that generates true probabilities per class, works with many token types, and provides lots of bells and whistles while being optimized for performance."
   s.email = "j@oasic.net"
   s.extra_rdoc_files = [

data/spec/nbayes_spec.rb

@@ -148,4 +148,14 @@ describe "NBayes" do
     end
   end
 
+  it "should dump to yaml string and load from yaml string" do
+    @nbayes.train( %w[a a a a], 'classA' ) 
+    @nbayes.train( %w[b b b b], 'classB' ) 
+    results = @nbayes.classify( ['b'] )
+    results['classB'].should >= 0.5
+    yml = @nbayes.dump(@nbayes)
+    @nbayes2 = NBayes::Base.new.load(yml)
+    results = @nbayes.classify( ['b'] )
+    results['classB'].should >= 0.5
+  end
 end

metadata

@@ -2,7 +2,7 @@
 name: nbayes
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors: 
 - oasic
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-06-27 00:00:00 Z
+date: 2012-07-13 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rspec
@@ -91,7 +91,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: -4266862066196834109
+      hash: 1596303061917544948
       segments: 
       - 0
       version: "0"