te_rex 0.0.5 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 17bee1b2969c5ea0885ee13cab910a660315c186
-  data.tar.gz: d68c27c7666231c8e1d3431a495d3f6f93b9a32d
+  metadata.gz: 41cbaf3893f705db1f366bdf247e0882971d860b
+  data.tar.gz: 909c02dd6578c76bdc5ec7ae4cab8f3b7cedd8d1
 SHA512:
-  metadata.gz: e7f412936cb643c27057eb2ba8f3f4c480e293ffa816f4f4d1724957d0a39710cc977e5e6076fdf0f3ea96cb29c19b99a88f873eafc5ebd5ecacdaa259c2eaf2
-  data.tar.gz: 440189ed0d1f93b98f6cd0121642b71c5013f71ae9804e10d4d4fdb455732e48c2448c9931c76fd019ad6e544fc49babdf2f2e40950cc431bcf64b33f12c7e41
+  metadata.gz: b9878b1e86ec13e17dd892abc4c615e1e369f692ea020413f121687af8b70c50a597557f7f36674f2a87464657edd62edd9f4ca62ab901fe3e042d98443180b8
+  data.tar.gz: 08a16cf36d77697253a56147fc3903f505ef5c09ebbfa7eb6f03474e44f87418a29a8c4640c8a46fb242cc441a39e193fa57bb9a8218653095eec9b3c519451e

data/lib/te_rex/bayes.rb

@@ -6,10 +6,15 @@ module TeRex
     class Bayes
 
       attr_accessor :category_counts, :total_words
+      attr_reader :messages
 
+      # categories = [{:tag => "Thing1", :msg => "Thing1 message"}, {:tag => "Thing2", :msg => "Thing2 message"}]
+      # initialize({:tag => "Refund", :msg => "You'll get a refund"}, {:tag => "Nonrefund", :msg => "You won't get a refund"})
       def initialize(*categories)
-        @clasif = Hash.new 
-        categories.each {|cat| @clasif[TeRex::Format.category_term(cat)] = Hash.new}
+        @clasif = Hash.new
+        @messages = Hash.new
+        categories.each {|cat| @clasif[TeRex::Format.category_term(cat[:tag])] = Hash.new}
+        categories.each {|cat| @messages[cat[:tag]] = cat[:msg]}
         @total_words = 0
         @category_counts = Hash.new(0)
       end
@@ -18,7 +23,7 @@ module TeRex
         category = TeRex::Format.category_term(ctgry)
         @category_counts[category] += 1
 
-        BayesData.index_frequency(text).each do |word, count| 
+        BayesData.index_frequency(text).each do |word, count|
           @clasif[category][word] ||= 0
           @clasif[category][word] += count
 
@@ -46,7 +51,8 @@ module TeRex
       end
 
       def classify(text)
-        (classifications(text).sort_by{|a| -a[1]})[0][0]
+        tag = (classifications(text).sort_by{|a| -a[1]})[0][0]
+        [tag, @messages[tag]]
       end
 
       def categories
@@ -59,7 +65,7 @@ module TeRex
         @clasif.each_pair do |term,val|
           cc = self.category_counts[term]
           train_ratio = (@total_words/cc).to_f
-          tmp << [(train_ratio > max_threshold), term, "description" => {"training_ratio" => "#{train_ratio}", "threshold" => "#{max_threshold}", "category_counts" => "#{cc}", "total_words" => "#{@total_words}"}]
+          tmp << [(train_ratio >= max_threshold), term, "description" => {"training_ratio" => "#{train_ratio}", "threshold" => "#{max_threshold}", "category_counts" => "#{cc}", "total_words" => "#{@total_words}"}]
         end
         tmp
       end

data/lib/te_rex/version.rb

@@ -1,3 +1,3 @@
 module TeRex
-  VERSION = "0.0.5"
+  VERSION = "0.0.10"
 end

data/lib/te_rex.rb

@@ -1,3 +1,10 @@
+#require 'simplecov'
+#SimpleCov.command_name 'mt'
+#
+#SimpleCov.start  do
+#  add_filter 'test/test_modules'
+#end
+
 require_relative "format/format"
 require_relative "format/corpus/brown_file"
 require_relative "format/corpus/basic_file"

data/test/test_helper.rb

@@ -0,0 +1,9 @@
+require 'simplecov'
+SimpleCov.command_name 'mt'
+
+SimpleCov.start  do
+  add_filter 'test/modules'
+
+  add_group 'Formatter', 'lib/format'
+  add_group 'Core', 'lib/te_lrex'
+end

data/test/test_modules/refund.rb

@@ -25,7 +25,6 @@ module TeRex
       "- 12 NOON HOTEL TIME DAY OF ARRIVAL TO AVOID BILLING OF 1NT ROOM AND TAX OR FORFEITURE OF DEPOSIT",
       "RESERVATIONS MUST BE CANCELLED BY 1600/4PM DAY OF ARRIVAL LOCAL HOTEL TIME TO AVOID A PENALTY OF ONE NIGHT ROOM AND TAX CHARGE. NOTE- INDIVIDUAL PLANS MAY VARY. PLEASE READ INDIVIDUAL DISPLAY FOR MORE INFORMATION. **50.00USD EARLY DEPARTURE FEE** .",
       "- RESERVATIONS BOOKED 8 DAYS OR MORE PRIOR TO ARRIVAL MUST BE CANCELLED 7 DAYS PRIOR TO ARRIVAL TO RECEIVE REFUND - RESERVATIONS BOOKED WITHIN 7 DAYS OF ARRIVAL FORFEIT DEPOSIT OF 3 NIGHTS ROOM AND TAX",
-      "ALL RESERVATIONS MUST BE CANCELLED 24 HOURS PRIOR TO HOST TIME UNLESS DEPOSIT REQUIRED IF THIS RESERVATION HAS BEEN MADE ELECTRONICALLY PLEASE CANCEL IT ELECTRONICALLY TO AVOID CONFUSION AND A NO SHOW BILL POLICY SUBJECT TO CHANGE .",
       "PLEASE REFER TO RATE DETAILS AT THE TIME OF BOOKING FOR CANCELLATION POLICY. PLEASE NOTE THAT POLICY IS SUBJECT TO CHANGES WITHOUT NOTICE."
     ]
   end

data/test/test_modules/unknown.rb

@@ -3,6 +3,7 @@ module TeRex
     UNKNOWN = [
       "The cancellation policy will be determined when the rate is validated.",
       "-CANCEL POLICY MAY VARY BY DAY OF WEEK AND SEASON. THE MOST ACCURATE CANCEL POLICY IS ADVISED DURING BOOKING PROCESS.  IN CASE OF A NO-SHOW THE CREDIT CARD WILL BE CHARGED ONE NIGHT STAY.  OUR SYSTEM ACKNOWLEDGES ALL PROPERLY CANCELED RESERVATIONS BY RETURNING A CANCELLATION NUMBER. DO NOT ASSUME YOUR RESERVATION IS CANCELED IF YOU HAVE NOT RECEIVED A CANCELLATION NUMBER IN YOUR PNR OR BOOKING FILE. IF YOU DO NOT RECEIVE A CANCELLATION NUMBER, PLEASE CALL THE CHOICE GDS DEPARTMENT AT 1-866-953-4570",
+      "CANCELLATION DEADLINES MAY VARY BY DATES OF ARRIVAL/ SPECIAL EVENTS OR BY RATE PLAN. PLEASE READ THE RATE RULES FOR YOUR RESERVATION FOR THE EXACT DEADLINE. FAILURE TO CANCEL WITHIN THE DEADLINE WILL RESULT IN A CANCELLATION FEE THAT COULD RANGE FROM 1 NIGHT UP TO THE TOTAL AMOUNT OF STAY. NON REFUNDABLE RATES CANNOT BE CANCELLED AND ARE SUBJECT TO FULL AMOUNT OF STAY PENALTY. -EARLY CHECKOUT POLICY -NO CHARGE FOR EARLY DEPARTURE AS LONG AS THE GUEST CHECKS OUT BY 12PM EASTERN STANDARD TIME -EXCEPTIONS MAY APPLY DURING SPECIAL EVENTS OR CONVENTIONS - SEE RATE RULES.",
       "-14JAN02 - END - CANCEL POLICIES VARY BY HOTEL.  SINCE A HOTEL CAN SET A CANCELLATION POLICY OF UP TO 30 DAYS IN ADVANCE, PLEASE REVIEW POLICY PRIOR TO BOOKING TO AVOID POSSIBLE CHARGE."
     ]
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: te_rex
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.10
 platform: ruby
 authors:
 - Joshua Bowles
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-23 00:00:00.000000000 Z
+date: 2014-11-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fast-stemmer
@@ -111,65 +111,59 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 0.10.1
 - !ruby/object:Gem::Dependency
-  name: pry-debugger
+  name: pry-rescue
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.2'
+        version: '1.4'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.3
+        version: 1.4.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.2'
+        version: '1.4'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.3
+        version: 1.4.1
 - !ruby/object:Gem::Dependency
-  name: pry-rescue
+  name: pry-stack_explorer
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.4'
+        version: '0.4'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.4.1
+        version: 0.4.9.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.4'
+        version: '0.4'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.4.1
+        version: 0.4.9.1
 - !ruby/object:Gem::Dependency
-  name: pry-stack_explorer
+  name: simplecov
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.4'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.4.9.1
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.4'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.4.9.1
+        version: '0'
 description: Simple text processing for small data sets.
 email:
 - jbowayles@gmail.com
@@ -190,12 +184,13 @@ files:
 - test/alpha_num_test.rb
 - test/bayes_data_test.rb
 - test/sparse_bayes_test.rb
+- test/test_helper.rb
 - test/test_modules/nonrefund.rb
 - test/test_modules/partrefund.rb
 - test/test_modules/refund.rb
 - test/test_modules/unknown.rb
 - test/trained_bayes_test.rb
-homepage: ''
+homepage: https://github.com/jbowles/te_rex
 licenses:
 - MIT
 metadata: {}
@@ -215,7 +210,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.1
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Basic NLP stuff for small data sets. Naive Bayes classification and corpora
@@ -224,6 +219,7 @@ test_files:
 - test/alpha_num_test.rb
 - test/bayes_data_test.rb
 - test/sparse_bayes_test.rb
+- test/test_helper.rb
 - test/test_modules/nonrefund.rb
 - test/test_modules/partrefund.rb
 - test/test_modules/refund.rb