fast_bayes 1.0.1 → 1.1.0

data/fast_bayes.gemspec

@@ -23,6 +23,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_runtime_dependency 'rice', '~> 2.1', '>= 2.1.0'
+  spec.add_runtime_dependency 'fast-stemmer', '~> 1.0', '>= 1.0.0'
 
   spec.add_development_dependency "bundler", "~> 1.13"
   spec.add_development_dependency "rake", "~> 10.0"

data/lib/fast_bayes.rb

@@ -1,7 +1,25 @@
 require "fast_bayes/version"
+require 'fast-stemmer'
+require "fast_bayes/fast_bayes"
 
 class FastBayes
 
-end
+  # Stem all words, this is a lot slower (TODO this fast)
+  def observe_better(data, label)
+    newdata = ""
+    data.split(" ").each do |i|
+      newdata += (i.stem + " ")
+    end
+    observe(newdata, label)
+  end
 
-require "fast_bayes/fast_bayes"
+  # Stem all words, this is a lot slower (TODO this fast)
+  def classify_better(data)
+    newdata = ""
+    data.split(" ").each do |i|
+      newdata += (i.stem + " ")
+    end
+    classify(newdata)
+  end
+
+end

data/lib/fast_bayes/version.rb

@@ -1,3 +1,3 @@
 class FastBayes
-  VERSION = "1.0.1"
+  VERSION = "1.1.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fast_bayes
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Chang Rajani
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-12-02 00:00:00.000000000 Z
+date: 2016-12-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -30,6 +30,26 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 2.1.0
+- !ruby/object:Gem::Dependency
+  name: fast-stemmer
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -108,6 +128,7 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- benchmarks/bench.rb
 - bin/console
 - bin/setup
 - ext/Makefile
@@ -115,11 +136,15 @@ files:
 - ext/fast_bayes/bayes.h
 - ext/fast_bayes/extconf.rb
 - ext/fast_bayes/main.cpp
+- ext/fast_bayes/stopwords.h
+- ext/fast_bayes/stopwords/en
 - ext/fast_bayes/wrapper.cpp
 - ext/test/catch.hpp
 - ext/test/classification_test.cpp
 - ext/test/data/20ng-test-all-terms.txt
 - ext/test/data/20ng-train-all-terms.txt
+- ext/test/data/r8-test-all-terms.txt
+- ext/test/data/r8-train-all-terms.txt
 - ext/test/test_helper.h
 - fast_bayes.gemspec
 - lib/fast_bayes.rb