diff --git a/README.md b/README.md index 41d1eb3..6cef0c4 100644 --- a/README.md +++ b/README.md @@ -58,7 +58,7 @@ training = [ dec_tree = DecisionTree::ID3Tree.new(labels, data, "not angry", color: :discrete, hunger: :continuous) dec_tree.train -decision = dec_tree.predict([7, "red"] +decision = dec_tree.predict([7, "red"]) puts "Predicted: #{decision} ... True decision: #{test.last}"; ```