From 92b9a86699a8760c16065bd529d19dc3d9304c2f Mon Sep 17 00:00:00 2001 From: Brian Underwood Date: Tue, 9 May 2017 16:12:38 -0400 Subject: [PATCH] De-sum 2 --- lib/decisiontree/id3_tree.rb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb index 9097ebf..8df1483 100755 --- a/lib/decisiontree/id3_tree.rb +++ b/lib/decisiontree/id3_tree.rb @@ -120,7 +120,7 @@ module DecisionTree index = attributes.index(attribute) values = data.map { |row| row[index] }.uniq - remainder = values.sort.sum do |val| + remainder = values.sort.inject(0, :+) do |val| classification = data.each_with_object([]) do |row, result| result << row.last if row[index] == val end