bae 0.0.9 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c28a60c92163259beddf8af99cd31357cf3a95a8
-  data.tar.gz: 34cc3ee332ec6f79d74e2ab04f8d60227c49c1a9
+  metadata.gz: ae80b08d7f073815336ca2f4b6d485e63cc64a16
+  data.tar.gz: 318820b928f8d66049ad5f329b69cc6c47a1bee3
 SHA512:
-  metadata.gz: 4f92cc52a40438b18bf543299345b1a9ae57443b53e8b8cae7169181a436dfc2e4c86627527239c6c0c29d99f0ff723c4f9272a994f4595d58c1f6d5c8acbd5c
-  data.tar.gz: dfca0d36849a088fdc5c60cb62c9256a7743c329d8acbbdba87cac34f0fb3de1195909598474283a12862a5a1b92688258189b47945330dcf8cc6df563b77e19
+  metadata.gz: 9299062efc70e820d4b491e7b7e257f6cba19f49968e55f3e6c05bbee6b6b7af2178bd829ca019238e8e3b9d654b8082c376e91169202f8c41d49c84a0fafb7a
+  data.tar.gz: c8d55c7517dc22c9ab937e73d523132c68f940ebb297f9f8798a9b8d736011fbdbcdf60e1fc8b3a71aa039b33d505765bb2a2a14e36489e686efe98303baa0be

data/lib/bae/classifier.rb

@@ -87,20 +87,13 @@ module Bae
     end
 
     def save_state(path)
-      state = {}
-      state['frequency_table'] = frequency_table
-      state['label_instance_count'] = label_instance_count
-      state['label_index'] = label_index
-      state['label_index_sequence'] = label_index_sequence
-      state['total_terms'] = total_terms
-
       ::File.open(::File.expand_path(path), 'w') do |handle|
-        handle.write(state.to_json)
+        handle.write(to_json)
       end
     end
 
-    def load_state(path)
-      state = ::JSON.parse(::File.read(::File.expand_path(path)))
+    def load_from_json(json)
+      state = ::JSON.parse(json)
 
       fail 'Missing frequency_table' unless state['frequency_table']
       fail 'Missing label_instance_count' unless state['label_instance_count']
@@ -117,6 +110,21 @@ module Bae
       finish_training!
     end
 
+    def load_state(path)
+      state_json = ::File.read(::File.expand_path(path))
+      load_from_json(state_json)
+    end
+
+    def to_json
+      state = {}
+      state['frequency_table'] = frequency_table
+      state['label_instance_count'] = label_instance_count
+      state['label_index'] = label_index
+      state['label_index_sequence'] = label_index_sequence
+      state['total_terms'] = total_terms
+      state.to_json
+    end
+
   private
 
     def calculate_likelihoods!

data/lib/bae/version.rb

@@ -1,3 +1,3 @@
 module Bae
-  VERSION = "0.0.9"
+  VERSION = "0.1.0"
 end

data/spec/lib/bae/classifier_spec.rb

@@ -47,6 +47,15 @@ describe ::Bae::Classifier do
     expect{ subject.classify(1337) }.to raise_error 'Training data must either be a string or hash'
   end
 
+  it "can dump the classifier as json" do
+    subject.train("positive", {"aaa" => 0, "bbb" => 1})
+    subject.train("negative", {"ccc" => 2, "ddd" => 3})
+
+    subject.finish_training!
+
+    expect(subject.to_json).to eq(state_json)
+  end
+
   it "can save the classifier state" do
     subject.train("positive", {"aaa" => 0, "bbb" => 1})
     subject.train("negative", {"ccc" => 2, "ddd" => 3})
@@ -63,7 +72,16 @@ describe ::Bae::Classifier do
     temp_file.unlink
   end
 
-  it "can correctly load a classifier state and correctly classify" do
+  it "can correctly load a classifier state from json and correctly classify" do
+    subject.load_from_json(state_json)
+
+    results = subject.classify({"aaa" => 1, "bbb" => 1})
+
+    expect(results["positive"]).to be_within(0.001).of(0.94117)
+    expect(results["negative"]).to be_within(0.001).of(0.05882)
+  end
+
+  it "can correctly load a classifier state from a file and correctly classify" do
     temp_file = ::Tempfile.new('some_state')
     temp_file.write(state_json)
     temp_file.rewind

metadata

@@ -1,19 +1,19 @@
 --- !ruby/object:Gem::Specification
 name: bae
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.1.0
 platform: ruby
 authors:
 - Garrett Thornburg
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2015-02-27 00:00:00.000000000 Z
+date: 2017-06-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.6'
   name: bundler
@@ -21,13 +21,13 @@ dependencies:
   type: :development
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ~>
+    - - "~>"
       - !ruby/object:Gem::Version
         version: '1.6'
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   name: rspec
@@ -35,13 +35,13 @@ dependencies:
   type: :development
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   name: rake
@@ -49,7 +49,7 @@ dependencies:
   type: :development
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
 description: Multinomial naive bayes classifier with a kick of java
@@ -59,7 +59,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- .gitignore
+- ".gitignore"
 - Gemfile
 - LICENSE.txt
 - README.md
@@ -90,17 +90,17 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project:
-rubygems_version: 2.2.2
+rubygems_version: 2.6.10
 signing_key:
 specification_version: 4
 summary: Multinomial naive bayes classifier with a kick of java