RedditPostClassifierBot 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 24f519dba9e63cfe4f5c8bdccc5eb9bb4da91133
-  data.tar.gz: 86691d72a1f186828759409dccab43da978df2a5
+  metadata.gz: dc702c8142af33798cd4105a225d66bb494de9d8
+  data.tar.gz: e4fbb1206147d9ba5f162e4d9f7c0e421a20d422
 SHA512:
-  metadata.gz: 3f2ea6b4823fa43c6e255bba4ef51d93adc76189c9ebddae9738d8637f258c916bae51c5ed3d19c8692809567a9907fa77009ce262ad0f15e00705bdaff16635
-  data.tar.gz: 5931d91769d21c412698d60d3e42fa16282a132cf586a0299f0c8fd023d6b80adec81c13efe7a3835b6ca54383c603bb2c9a8985af6fe49b2ee8ea0f174df01a
+  metadata.gz: c200ef8b3c100d43041c2cebe2f805ea2e75bcd666058acc06eaf4a23889c0ace7c5d3f116469b4c27e11242d06e8a1cec49d823a0d000bef53530ca2f838887
+  data.tar.gz: ea1839df63d9437e28174d258ede4783bd904f76a6cd128f2f49c9213bf3557ecb3a40a016e308e13168ffebfe79cbca09ecbed56986223e50f06a963c7abfee

data/lib/RedditPostClassifierBot.rb

@@ -16,4 +16,8 @@ module RedditPostClassifierBot
   def self.classify_posts(path)
     classifier.fetch_and_classify path
   end
+
+  def trained_classifier!
+    NBayesClassifier.trained? ? classifier : train_classifier
+  end
 end

data/lib/RedditPostClassifierBot/nbayes_classifier.rb

@@ -2,6 +2,10 @@ module RedditPostClassifierBot
   class NBayesClassifier
     NBAYES_FILE = ENV.fetch "NBAYES_FILE_PATH", "./RPCB-nbayes.yml"
 
+    def self.trained?
+      File.exists?(NBAYES_FILE) && File.read(NBAYES_FILE).chomp.size > 100
+    end
+
     def initialize
       @nbayes = NBayes::Base.new
     end

data/lib/RedditPostClassifierBot/version.rb

@@ -1,3 +1,3 @@
 module RedditPostClassifierBot
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: RedditPostClassifierBot
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Diego Salazar