dkam/decisiontree
1
0
Fork 0
mirror of https://github.com/dkam/decisiontree.git synced 2025-12-28 07:04:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #9 from marshallshen/master

Browse Source 
Enhance documentation
This commit is contained in:
Ilya Grigorik
2013-03-22 20:52:20 -07:00
parent 7bf93a3a5e 5fa6d1c181
commit 33ea3f20fa
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
README.md
View File

@@ -41,6 +41,25 @@ decision = dec_tree.predict([37, 'sick'])
puts "Predicted: #{decision} ... True decision: #{test.last}"; puts "Predicted: #{decision} ... True decision: #{test.last}";
# => Predicted: sick ... True decision: sick # => Predicted: sick ... True decision: sick
# Specify type ("discrete" or "continuous") in the training data
labels = ["hunger", "color"]
training = [
[8, "red", "angry"],
[6, "red", "angry"],
[7, "red", "angry"],
[7, "blue", "not angry"],
[2, "red", "not angry"],
[3, "blue", "not angry"],
[2, "blue", "not angry"],
[1, "red", "not angry"]
]
dec_tree = DecisionTree::ID3Tree.new(labels, data, "not angry", color: :discrete, hunger: :continuous)
dec_tree.train
decision = dec_tree.predict([7, "red"]
puts "Predicted: #{decision} ... True decision: #{test.last}";
``` ```
## License ## License