apriori-ruby 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d476a1f153c942bce7b6ab5bcb6e3482547fa01b
-  data.tar.gz: 0cd9fe4e61335cb42e00ee7934ede7d8cd896658
+  metadata.gz: 8220249daec5efad651f9ccf7907cdd45b069969
+  data.tar.gz: 2b443b21b6b270ec7e042f29fb4a7b66a502b1c5
 SHA512:
-  metadata.gz: 945a4ef7009955375496c5666133a7e56929b33171a7336e0614160442793db8874367c55c5d1a58d7890e150f0fc99adbf680a162886d6c536d72f078b4296f
-  data.tar.gz: 141573132f47069e3504a86be9b68d0cd3765204894b37d069afe1768a5d0de9a41967e32bea4b785790b2c11f7e2458d15bc3519dc09cdb927aa04befd6cddc
+  metadata.gz: ef373588718fda497258e5b09c395c2d46cd4122f611d34e130efc56ff7810933a30a45c65440d7813ae6ed3aca160b9b03ce05e9540071256dac1896901ab08
+  data.tar.gz: e0fcdb9c5ebf93ef5bb5a70d54c5a4a831c66c4d4726cff8447e7a238c67ac8f08ba91094ce4b51bf616e3664a82e163fcb1188ba88b9b43941c0d57ea04405a

data/.rspec

@@ -1,4 +1,3 @@
 --color
---warnings
 --format documentation
 --require spec_helper

data/README.md

@@ -16,7 +16,7 @@ gem 'apriori-ruby'
 
 Sample Usage
 ```ruby
-  test_data = {t1: [1,2,3,4], t2:[1,2,4,5], t3: [2,3,4,5]}
+  test_data = [[1,2,3,4], [1,2,4,5], [2,3,4,5]]
   item_set = Apriori::ItemSet.new(test_data)
   support = 50
   confidence = 60

data/lib/apriori/item_set.rb

@@ -1,30 +1,30 @@
 module Apriori
   class ItemSet
-    attr_reader :data_set, :min_support, :min_confidence
+    attr_reader :transactions, :min_support, :min_confidence
 
-    def initialize(data_set)
-      @data_set = data_set
+    def initialize(transactions)
+      @transactions = transactions
     end
 
     def mine(min_support=0, min_confidence=0)
-      @min_support, @min_confidence = min_support, min_confidence
+      @min_support, @min_confidence, @candidates = min_support, min_confidence, nil
       create_frequent_item_sets(min_support)
       create_association_rules(min_confidence)
     end
 
     def create_frequent_item_sets min_support
       @min_support = min_support
-      make_item_sets unless frequent_item_sets.any?
+      make_item_sets
       frequent_item_sets
     end
 
     def create_association_rules min_confidence
       make_association_rules unless frequent_item_sets.empty?
-      association_rules.select{ |name, confidence| confidence >= min_confidence }
+      association_rules.select{ |name, rule| rule >= min_confidence }
     end
 
     def support item
-      (count_frequency(item).to_f / data_set.size) * 100
+      (count_frequency(item).to_f / transactions.size) * 100
     end
 
     def confidence set1, set2
@@ -32,18 +32,20 @@ module Apriori
     end
 
     def count_frequency set
-      data_set.map do |transaction, items|
+      transactions.map do |items|
         contains_all?(items, set)
-      end.reject {|item| item == false }.size
+      end.reject { |item| item == false }.size
     end
 
     private
-    attr_reader :candidates
+
+    def candidates
+      @candidates ||= transactions.flatten.uniq.map { |item| [item] }
+    end
 
     def make_item_sets
-      @candidates = initial_data_set
       while candidates.any?
-        list = List.new(reject_candidates)
+        list = List.new(new_candidates)
         frequent_item_sets << list
         @candidates = list.make_candidates
       end
@@ -74,12 +76,9 @@ module Apriori
       set.to_set.superset? subset.to_set
     end
 
-    def reject_candidates
-      candidates.reject{|item| support(item) < min_support}
+    def new_candidates
+      candidates.reject{ |item| support(item) < min_support }
     end
 
-    def initial_data_set
-      @initial_data_set ||= @data_set.values.flatten.uniq.map{|item| [item]}
-    end
   end
 end

data/lib/apriori/version.rb

@@ -1,3 +1,3 @@
 module Apriori
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end

data/spec/factories/apriori/apriori.rb

@@ -3,12 +3,11 @@ FactoryGirl.define do
     initialize_with {new(FactoryGirl.build(:sample_data))}
   end
 
-  factory :sample_data, class: Hash do
-    t1 ['Mango', 'Onion', 'Nintendo', 'Keychain', 'Eggs', 'Yoyo']
-    t2 ['Doll', 'Onion', 'Nintendo', 'Keychain', 'Eggs', 'Yoyo']
-    t3 ['Mango', 'Apple', 'Keychain', 'Eggs']
-    t4 ['Mango', 'Umbrella', 'Corn', 'Keychain', 'Yoyo']
-    t5 ['Corn', 'Onion', 'Onion', 'Keychain', 'Icecream', 'Eggs']
-    initialize_with { attributes }
+  factory :sample_data, class: Array do
+    initialize_with { [['Mango', 'Onion', 'Nintendo', 'Keychain', 'Eggs', 'Yoyo'],
+    ['Doll', 'Onion', 'Nintendo', 'Keychain', 'Eggs', 'Yoyo'],
+    ['Mango', 'Apple', 'Keychain', 'Eggs'],
+    ['Mango', 'Umbrella', 'Corn', 'Keychain', 'Yoyo'],
+    ['Corn', 'Onion', 'Onion', 'Keychain', 'Icecream', 'Eggs']] }
   end
 end

data/spec/lib/apriori/item_set_spec.rb

@@ -33,14 +33,14 @@ describe Apriori::ItemSet do
 
   context '#create_frequent_item_sets' do
     it 'creates frequent item sets for a given support' do
-      @set = Apriori::ItemSet.new({:t1 => ['1','2','3'], :t2 => ['1','2','4'], :t3 => ['1','4','5']})
+      @set = Apriori::ItemSet.new([['1','2','3'], ['1','2','4'], ['1','4','5']])
       expect(@set.create_frequent_item_sets(60).first.sets).to eql([['1'],['2'],['4']])
     end
   end
 
   context '#create_association_rules' do
     it 'creates association rules for all combinations' do
-      @set = Apriori::ItemSet.new({:t1 => ['1','2','3'], :t2 => ['1','2','4'], :t3 => ['1','4','5']})
+      @set = Apriori::ItemSet.new([['1','2','3'], ['1','2','4'], ['1','4','5']])
       @set.create_frequent_item_sets(60)
       expect(@set.create_association_rules(60)).to eql({"1=>2"=>66.66666666666666, "2=>1"=>100.0, "1=>4"=>66.66666666666666, "4=>1"=>100.0})
     end

metadata

@@ -1,55 +1,55 @@
 --- !ruby/object:Gem::Specification
 name: apriori-ruby
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors:
 - Bryan Mulvihill
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-10-06 00:00:00.000000000 Z
+date: 2015-07-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ~>
       - !ruby/object:Gem::Version
         version: '1.7'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ~>
       - !ruby/object:Gem::Version
         version: '1.7'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ~>
       - !ruby/object:Gem::Version
         version: '10.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ~>
       - !ruby/object:Gem::Version
         version: '10.0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - '>='
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - '>='
       - !ruby/object:Gem::Version
         version: '0'
 description: 
@@ -59,11 +59,9 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".gitignore"
-- ".rspec"
-- ".ruby-version"
+- .gitignore
+- .rspec
 - Gemfile
-- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
@@ -86,17 +84,17 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">="
+  - - '>='
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">="
+  - - '>='
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.0.14
 signing_key: 
 specification_version: 4
 summary: Ruby implementation of Apriori Algorithm

data/.ruby-version

@@ -1 +0,0 @@
-2.1.2

data/Gemfile.lock

@@ -1,46 +0,0 @@
-PATH
-  remote: .
-  specs:
-    apriori (0.0.1)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    activesupport (4.1.5)
-      i18n (~> 0.6, >= 0.6.9)
-      json (~> 1.7, >= 1.7.7)
-      minitest (~> 5.1)
-      thread_safe (~> 0.1)
-      tzinfo (~> 1.1)
-    diff-lcs (1.2.5)
-    factory_girl (4.4.0)
-      activesupport (>= 3.0.0)
-    i18n (0.6.11)
-    json (1.8.1)
-    minitest (5.4.0)
-    rake (10.3.2)
-    rspec (3.0.0)
-      rspec-core (~> 3.0.0)
-      rspec-expectations (~> 3.0.0)
-      rspec-mocks (~> 3.0.0)
-    rspec-core (3.0.4)
-      rspec-support (~> 3.0.0)
-    rspec-expectations (3.0.4)
-      diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.0.0)
-    rspec-mocks (3.0.4)
-      rspec-support (~> 3.0.0)
-    rspec-support (3.0.4)
-    thread_safe (0.3.4)
-    tzinfo (1.2.2)
-      thread_safe (~> 0.1)
-
-PLATFORMS
-  ruby
-
-DEPENDENCIES
-  apriori!
-  bundler (~> 1.7)
-  factory_girl
-  rake (~> 10.0)
-  rspec