dkam/decisiontree
1
0
Fork 0
mirror of https://github.com/dkam/decisiontree.git synced 2025-12-27 22:54:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #24 from lukeasrodgers/fix-readme

Browse Source 
Fix examples in README.md
This commit is contained in:
Ilya Grigorik
2014-09-19 13:24:04 -07:00
parent 6fb6d84b6a 5412430e95
commit eaa9a2f5d7
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
README.md
View File

@@ -37,8 +37,9 @@ training = [
dec_tree = DecisionTree::ID3Tree.new(attributes, training, 'sick', :continuous)
dec_tree.train
decision = dec_tree.predict([37, 'sick'])
puts "Predicted: #{decision} ... True decision: #{test.last}";
test = [37, 'sick']
decision = dec_tree.predict(test)
puts "Predicted: #{decision} ... True decision: #{test.last}"
# => Predicted: sick ... True decision: sick
@@ -55,11 +56,14 @@ training = [
[1, "red", "not angry"]
]
dec_tree = DecisionTree::ID3Tree.new(labels, data, "not angry", color: :discrete, hunger: :continuous)
dec_tree = DecisionTree::ID3Tree.new(labels, training, "not angry", color: :discrete, hunger: :continuous)
dec_tree.train
decision = dec_tree.predict([7, "red"])
puts "Predicted: #{decision} ... True decision: #{test.last}";
test = [7, "red", "angry"]
decision = dec_tree.predict(test)
puts "Predicted: #{decision} ... True decision: #{test.last}"
# => Predicted: angry ... True decision: angry
```
## License