inferx 0.2.0 → 0.2.1

data/lib/inferx/category.rb

@@ -71,6 +71,14 @@ class Inferx
       end
     end
 
+    # Prepare to enhance the training data. Use for high performance.
+    #
+    # @yield [train] process something
+    # @yieldparam [Proc] train enhance the training data giving words
+    def ready_to_train(&process)
+      train(aggregate(&process))
+    end
+
     # Attenuate the training data giving words.
     #
     # @param [Array<String>] words an array of words
@@ -98,6 +106,14 @@ class Inferx
       end
     end
 
+    # Prepare to attenuate the training data giving words.
+    #
+    # @yield [untrain] process something
+    # @yieldparam [Proc] untrain attenuate the training data giving words
+    def ready_to_untrain(&process)
+      untrain(aggregate(&process))
+    end
+
     # Get effectively scores for each word.
     #
     # @param [Array<String>] words an array of words
@@ -123,5 +139,11 @@ class Inferx
         hash
       end
     end
+
+    def aggregate
+      all = []
+      yield lambda { |items| all += items }
+      all
+    end
   end
 end

data/lib/inferx/version.rb

@@ -1,3 +1,3 @@
 class Inferx
-  VERSION = '0.2.0'
+  VERSION = '0.2.1'
 end

data/spec/inferx/category_spec.rb

@@ -122,6 +122,19 @@ describe Inferx::Category, '#train' do
   end
 end
 
+describe Inferx::Category, '#ready_to_train' do
+  it 'calls #train with the words to train block' do
+    category = described_class.new(redis_stub, :red, 2)
+    category.should_receive(:train).with(%w(word1 word2 word3))
+
+    category.ready_to_train do |train|
+      train[%w(word1)]
+      train[%w(word2)]
+      train[%w(word3)]
+    end
+  end
+end
+
 describe Inferx::Category, '#untrain' do
   it 'calls Redis#zincrby, Redis#zremrangebyscore and Redis#hincrby' do
     redis = redis_stub do |s|
@@ -177,6 +190,19 @@ describe Inferx::Category, '#untrain' do
   end
 end
 
+describe Inferx::Category, '#ready_to_untrain' do
+  it 'calls #untrain with the words to untrain block' do
+    category = described_class.new(redis_stub, :red, 2)
+    category.should_receive(:untrain).with(%w(word1 word2 word3))
+
+    category.ready_to_untrain do |untrain|
+      untrain[%w(word1)]
+      untrain[%w(word2)]
+      untrain[%w(word3)]
+    end
+  end
+end
+
 describe Inferx::Category, '#scores' do
   it 'calls Redis#zscore' do
     redis = redis_stub do |s|

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: inferx
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-18 00:00:00.000000000 Z
+date: 2012-10-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis
@@ -87,7 +87,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.21
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: Naive Bayes classifier, the training data on Redis