bugfix: after adding @stemmer on Classifier/LSI the objects couldnt b…

…e marshalled.

lib/classifier/bayes.rb

@@ -131,6 +131,15 @@ def add_category(category)
 	end
 
 	alias append_category add_category
+
+  def marshal_dump
+    [@categories, @total_words, @options ]
+  end
+
+  def marshal_load(data)
+    @categories, @total_words, @options = data
+  end
+
 end
 
 end

lib/classifier/lsi.rb

@@ -295,6 +295,17 @@ def highest_ranked_stems( doc, count=3 )
       return top_n.collect { |x| @word_list.word_for_index(arr.index(x))}
     end
 
+    def marshal_dump
+      [ @auto_rebuild, @word_list, @items, @version, @built_at_version,
+        @options,
+      ]
+    end
+
+    def marshal_load(data)
+      @auto_rebuild, @word_list, @items, @version, @built_at_version,
+        @options = data
+    end
+
     private
     def build_reduced_matrix( matrix, cutoff=0.75 )
       # TODO: Check that M>=N on these dimensions! Transpose helps assure this

luisparravicini-classifier.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{luisparravicini-classifier}
-  s.version = "1.3.7"
+  s.version = "1.3.8"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Yury Korolev"]

test/bayes/bayesian_test.rb

@@ -55,4 +55,14 @@ def test_case_insensitive
 	  assert_equal c.classifications("ХОРОШО"), c.classifications("хорошо")
 	  assert_equal c.classifications("плОХО"), c.classifications("плохо")
   end
+
+  def test_serialize
+    txt = "this can be serialized"
+    b = Classifier::Bayes.new(:categories => ['Interesting', 'Uninteresting'])
+    b.train_interesting(txt)
+    b.train_uninteresting("really uninteresting")
+
+    b2 = Marshal::load(Marshal::dump(b))
+    assert_equal b.classify(txt), b2.classify(txt)
+  end
 end