mirror of
https://github.com/dkam/decisiontree.git
synced 2025-12-28 15:14:52 +00:00
fix return values; add missing files (doh!)
This commit is contained in:
3
test/helper.rb
Normal file
3
test/helper.rb
Normal file
@@ -0,0 +1,3 @@
|
||||
require 'rubygems'
|
||||
require 'spec'
|
||||
require 'lib/decisiontree'
|
||||
@@ -8,10 +8,10 @@ describe DecisionTree::ID3Tree do
|
||||
[1,0,1],
|
||||
[0,1,0]
|
||||
]
|
||||
|
||||
|
||||
dec_tree = DecisionTree::ID3Tree.new(labels, data, 1, :discrete)
|
||||
dec_tree.train
|
||||
|
||||
|
||||
dec_tree.predict([1,0]).should == 1
|
||||
dec_tree.predict([0,1]).should == 0
|
||||
end
|
||||
|
||||
Reference in New Issue
Block a user