classifier 1.4.0 → 1.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c20227e9f55f35fe93f2cbb6e5fc132127159973da74d7af68d1680a25021e73
-  data.tar.gz: ab8146db131a32b455b4e9b47dc3db7964e1e1a3bf9c69e953fdc6e6305eab89
+  metadata.gz: 100ebc0413bceeeb5148d4fed13de9cf9c9400016fc9f005d0d6a8f440ecf361
+  data.tar.gz: d8ff488e9fd84e28539b2546a1bb338deeecde8f24b0bcf312c04e2866137968
 SHA512:
-  metadata.gz: 2d024e1d46b3529f7e328110e8b59161e0f6fa0d738b3de0d973073974ad69fd7d3bfb529d5a1b827a444266c14f5dff37630af2e522522d07d74f18324064eb
-  data.tar.gz: 12fa59e6fc0f3e5ffb4fe1dc838b00ca529e630875729c78a3a0847bfdb6367856799c0f95831ecd75dbb36d0f4fad7fea2a6082083a965a5529d8bbf75a1542
+  metadata.gz: f41ee004b5d4c0a6cb18a9694fe1d69dcfdf0a7d4e3598086acf0c1bd786db43d5ef2a5b2d8438f1097eef0031bae7d8a7360a36fd0a7b28acad8f25fffb69a5
+  data.tar.gz: acefed9a7917ba3b1cdae5c23e0cb708533cf09b740a3b6d546ceabbd0d980abbfcce177fc89737a3a8afab3169eaa7f8e9ae99b8f870642a7ee68163064311d

data/lib/classifier/bayes.rb

@@ -139,5 +139,23 @@ module Classifier
     end
 
     alias append_category add_category
+
+    #
+    # Allows you to remove categories from the classifier.
+    # For example:
+    #     b.remove_category "Spam"
+    #
+    # WARNING: Removing categories from a trained classifier will
+    # result in the loss of all training data for that category.
+    # Make sure you really want to do this before calling this method.
+    def remove_category(category)
+      category = category.prepare_category_name
+      raise StandardError, "No such category: #{category}" unless @categories.key?(category)
+
+      @categories.delete(category)
+      @category_counts.delete(category)
+      @category_word_count.delete(category)
+      @total_words -= @category_word_count[category].to_i
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: classifier
 version: !ruby/object:Gem::Version
-  version: 1.4.0
+  version: 1.4.1
 platform: ruby
 authors:
 - Lucas Carlson