decision_tree 0.1.0

data/test/test_decision_tree.rb

@@ -0,0 +1,25 @@
+#The MIT License
+
+###Copyright (c) 2007 Ilya Grigorik <ilya AT fortehost DOT com>
+
+require File.dirname(__FILE__) + '/test_helper.rb'
+require 'decision_tree'
+
+class TestDecisionTree < Test::Unit::TestCase
+
+  def setup
+    @labels = %w(sun rain)
+    @data = [
+        [1, 0, 1],
+        [0, 1, 0]
+    ]
+  end
+  
+  def test_truth
+    dec_tree = DecisionTree::ID3Tree.new(@labels, @data, 1, :discrete)
+    dec_tree.train
+    
+    assert 1, dec_tree.predict([1, 0])
+    assert 0, dec_tree.predict([0, 1])
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,2 @@
+require 'test/unit'
+require File.dirname(__FILE__) + '/../lib/decision_tree'

metadata

@@ -0,0 +1,56 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.0
+specification_version: 1
+name: decision_tree
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+date: 2007-04-05 00:00:00 +00:00
+summary: ID3-based implementation of the M.L. Decision Tree algorithm
+require_paths: 
+  - lib
+email: ilya <at> fortehost.com
+homepage: http://decision_tree.rubyforge.org
+rubyforge_project: decision_tree
+description: ID3-based implementation of the M.L. Decision Tree algorithm
+autorequire: 
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+    - 
+      - ">"
+      - !ruby/object:Gem::Version 
+        version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+  - Ilya Grigorik
+files: 
+  - CHANGELOG.txt
+  - History.txt
+  - Manifest.txt
+  - README.txt
+  - Rakefile
+  - lib/decision_tree.rb
+  - lib/decision_tree/version.rb
+  - lib/decision_tree/id3_tree.rb
+  - setup.rb
+  - examples/continuous-id3.rb
+  - examples/discrete-id3.rb
+  - examples/data/continuous-test.txt
+  - examples/data/continuous-training.txt
+  - examples/data/discrete-test.txt
+  - examples/data/discrete-training.txt
+  - test/test_helper.rb
+  - test/test_decision_tree.rb
+test_files: []
+rdoc_options: []
+extra_rdoc_files: []
+executables: []
+extensions: []
+requirements: []
+dependencies: []