From 5fa6d1c181dbabcef1d6ec92a1c916905db74c27 Mon Sep 17 00:00:00 2001 From: Marshall Shen Date: Fri, 22 Mar 2013 13:01:24 -0500 Subject: [PATCH] enhance documentation --- README.md | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/README.md b/README.md index 25ecb71..41d1eb3 100644 --- a/README.md +++ b/README.md @@ -41,6 +41,25 @@ decision = dec_tree.predict([37, 'sick']) puts "Predicted: #{decision} ... True decision: #{test.last}"; # => 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