From 5412430e95bdd250c744bd853fec63441bf6e5ff Mon Sep 17 00:00:00 2001 From: Luke Rodgers Date: Fri, 19 Sep 2014 15:30:53 -0400 Subject: [PATCH] Fix examples in README.md `test` variables were missing, second example referenced missing `data` variable. --- README.md | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 6cef0c4..bf0abec 100644 --- a/README.md +++ b/README.md @@ -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