huginn_naive_bayes_agent 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d109e936add2768720a9449441a5cc178e0410c3
-  data.tar.gz: 6153de43815a514561531248aae4ac0a2a15224d
+  metadata.gz: 6d1a1c1469bb599fe8a5a42a94e89fd6206e519d
+  data.tar.gz: d2ce17b2c8471199add13b73c92047cb84881c4a
 SHA512:
-  metadata.gz: df773ad2a00bce7947a47ecd3bc9ebf0c3b0c647c92912a6ef86c24fb84b20dee537f1b81d18a1a5b3fb47a9b1203022be2f5061091adb19797130b921953053
-  data.tar.gz: 62b156d2b093b51e794a1518363d35cd43f4ff9b34fa39a42344d6eeebb030f2cd26fc1a2f75109b20a54be737c2558e3fd11a404bae113d8b7a47f6347df588
+  metadata.gz: cb4ae84beb82c52900b99d21c9ca3d542f9e99f1bee34794ef1bb92e6191559d89f4afa6c8a7f5cbf5918e1967df8dadd0c89c90bd9d8f7ffd32382017a4018e
+  data.tar.gz: a4c1e57328ddc1fef840f73aeeeb5072fc251b2d5fe03e649c175e43b46515775f0fa74336d3bbd1964280ee83276e6a8a0d431a786b7c59998c1ec188cdcc07

data/lib/huginn_naive_bayes_agent/naive_bayes_agent.rb

@@ -14,7 +14,7 @@ module Agents
         * `nb_content` for the content used for classification, space separated. 
         * `nb_cats` for the classification categories, space separated.
         
-      If `nb_cats` is empty, then the content from `nb_content` will be classified according to the training data. The categories will be added to `nb_cats` and then a new event is created with that payload.
+      If `nb_cats` is empty or contains `=class`, then the content from `nb_content` will be classified according to the training data. The categories will be added to `nb_cats` and then a new event is created with that payload.
       
       However, if `nb_cats` is already populated, then the content from `nb_content` will be used as training data for the categories listed in `nb_cats`. For instance, say `nb_cats` consists of `trees`. Then `nb_content` will be used as training data for the category `trees`. The data is saved to the agent memory. 
       
@@ -59,7 +59,7 @@ module Agents
     def receive(incoming_events)
       incoming_events.each do |event|
         nbayes = load(memory['data'])
-        if event.payload['nb_cats'].length != 0
+        if event.payload['nb_cats'].length > 0 and not event.payload['nb_cats'].include?("=class")
           cats = event.payload['nb_cats'].split(/\s+/)
           if cats[0] == "=loadYML"
             memory['data'] = event.payload['nb_content']
@@ -84,7 +84,7 @@ module Agents
         else
           result = nbayes.classify(event.payload['nb_content'].split(/\s+/))
           if interpolated['min_value'].to_f == 1
-            result.max_class
+            event.payload['nb_cats'] << (event.payload['nb_cats'].length == 0 ? result.max_class : " "+result.max_class)
           else
             result.each do |cat, val|
               if val > interpolated['min_value'].to_f
@@ -113,24 +113,5 @@ module Agents
       nbayes
     end
 
-#    def purge_less_than(x)
-#      remove_list = {}
-#      nbayes = load(memory['data'])
-#      nbayes.vocab.each do |token|
-#        if nbayes.data.purge_less_than(token, x)
-#          # print "removing #{token}\n"
-#          remove_list[token] = 1
-#        end
-#      end  # each vocab word
-#      remove_list.keys.each {|token| nbayes.vocab.delete(token) }
-#      memory['data'] = YAML.dump(nbayes)
-#    end
-
-    # Delete an entire category from the classification data
-#    def delete_category(category)
-#      nbayes = load(memory['data'])
-#      nbayes.data.delete_category(category)
-#      memory['data'] = YAML.dump(nbayes)
-#    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: huginn_naive_bayes_agent
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Noah Greenstein
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-04-25 00:00:00.000000000 Z
+date: 2017-05-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler