fBayes 0.2.1 → 0.2.2

data/VERSION

@@ -1 +1 @@
-0.2.1
+0.2.2

data/fBayes.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = "fBayes"
-  s.version = "0.2.1"
+  s.version = "0.2.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Ivan Acosta-Rubio"]

data/lib/data/cdc/classifier.rb

@@ -11,8 +11,8 @@ puts "*" * 100
 puts "CDC CLASSIFIER"
 puts "*" * 100
 
-categories_path = Dir[ FBayesDir.root + "/cdc/data/*"]
-categories = categories_path.map{|path| path.gsub(FBayesDir.root + "/cdc/data/", "")}
+categories_path = Dir[ FBayesDir.cdc_data + "*"]
+categories = categories_path.map{|path| path.gsub(FBayesDir.cdc_data, "")}
 @cdc_classifier = Classifier::Bayes.new 
 
 categories.each do |c|
@@ -24,12 +24,9 @@ total_size = 0
 
 categories.each do |category|
   method = "train_" + category.downcase
-  data = File.read( FBayesDir.root + "/cdc/data/#{category}")
-
-
+  data = File.read( FBayesDir.cdc_data + category.to_s )
   puts category +": " + data.size.to_s
   total_size += data.size.to_i
-
   @cdc_classifier.send(method, data)
 end
 

data/lib/data/sr24/classifier.rb

@@ -4,9 +4,9 @@ puts "*" * 100
 puts "SR24 CLASSIFIER:"
 puts "*" * 100
 
-categories_path =  Dir[ FBayesDir.root + "/data/*"]
+categories_path =  Dir[ FBayesDir.sr24_data + "*"]
 
-@categories = categories_path.map{|path| path.gsub(FBayesDir.root + "/data/", "")}
+@categories = categories_path.map{|path| path.gsub(FBayesDir.sr24_data, "")}
 
 @sr24_classifier = Classifier::Bayes.new 
 
@@ -19,7 +19,7 @@ total_size = 0
 
 @categories.each do |category|
   method = "train_" + category.downcase
-  data = File.read( FBayesDir.root + "data/#{category}")
+  data = File.read( FBayesDir.sr24_data + category.to_s)
   puts category +": " + data.size.to_s
   @sr24_classifier.send(method, data)
   total_size += data.size.to_i

data/lib/fBayes.rb

@@ -4,6 +4,15 @@ module FBayesDir
   def self.root
     Gem::Specification.find_by_name("fBayes").gem_dir + "/lib/data/"
   end
+
+  def self.cdc_data
+    self.root + "/cdc/data/"
+  end
+
+  def self.sr24_data
+
+    self.root + "/sr24/data/"
+  end
 end
 
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fBayes
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
   prerelease: 
 platform: ruby
 authors:
@@ -184,7 +184,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2836690683508422044
+      hash: 778778263631434492
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: