examples/large_file_test.rb @@ -1,7 +1,13 @@ These are things todo: -+ Clean up licensing everywhere +* refactor find_itemsets to be find_association_rules + * Clean up documentation + * README.txt + * apriori.rb + * create an example file * make a ruby gem + ++ Clean up licensing everywhere + get the typical install to work -* clean up the rakefile ++ clean up the rakefile TODO.txt @@ -12,7 +12,8 @@ void Init_mytest(); // Prototype for our method 'test1' - methods are prefixed by 'method_' here VALUE method_do_test_apriori(VALUE self); //VALUE method_test_hash_ap(VALUE self, VALUE filein, VALUE fileout, VALUE opts); -VALUE method_find_association_rules(VALUE self, VALUE filein, VALUE fileout, VALUE opts); +//VALUE method_find_association_rules(VALUE self, VALUE filein, VALUE fileout, VALUE opts); + VALUE method_ap_do_apriori(VALUE self, VALUE rargv); // The initialization method for this module ext/Apriori.c @@ -39,7 +39,7 @@ module Apriori # # this will probably become refactored out and then use a block b/c most # of the setup is always the same - def self.find_itemsets(input, opts={}) + def self.find_association_rules(input, opts={}) args = [] # create the input file lib/apriori.rb @@ -9,12 +9,12 @@ class TestApriori < Test::Unit::TestCase def test_truth input = File.join(FIXTURES_DIR + "/market_basket_string_test.txt") - assert rules = Apriori.find_itemsets(input) + assert rules = Apriori.find_association_rules(input) end def test_calling_all_the_options input = File.join(FIXTURES_DIR + "/market_basket_string_test.txt") - assert rules = Apriori.find_itemsets(input, + assert rules = Apriori.find_association_rules(input, :min_items => 2, :max_items => 5, :min_support => 1, @@ -30,7 +30,7 @@ class TestApriori < Test::Unit::TestCase %w{apple cheese}, %w{apple doritos} ] - rules = Apriori.find_itemsets(transactions, + rules = Apriori.find_association_rules(transactions, :min_items => 2, :max_items => 5, :min_support => 1, @@ -43,4 +43,26 @@ class TestApriori < Test::Unit::TestCase # puts rules.join("\n") end + def test_returning_no_results + # currently the application exits. thats no good! just give a message + transactions = [ %w{beer doritos}, + %w{apple cheese}, + %w{beer doritos}, + %w{apple cheese}, + %w{apple cheese}, + %w{apple doritos} ] + + # crazy numbers that will never return any rules + rules = Apriori.find_association_rules(transactions, + :min_items => 5, + :max_items => 5, + :min_support => 100, + :max_support => 100, + :min_confidence => 20) + + assert_equal 0, rules.size + + # also, make notes that if you get more rules, a common thing to do is to reduce the min_support to a very small float + end + end test/unit/test_apriori.rb d33fd5e94e18c4c5b0f4cbad2592fdef37914dea Nate Murray nate@natemurray.com http://github.com/jashmenn/apriori/commit/486072f744ee982911daf9ff0b526ba82fc0ab5a 486072f744ee982911daf9ff0b526ba82fc0ab5a 2008-09-02T18:41:14-07:00 2008-09-02T18:41:14-07:00 changed find_itemsets -> find_association_rules added a test for large files 9fb00a7d12ecbf26fd01f91fdd76a28e5947dbd6 Nate Murray nate@natemurray.com