Merge pull request #34 from cheerfulstoic/patch-2

De-sum
igrigorik · May 9, 2017 · 169451e · 169451e
2 parents 846448a + 92b9a86
commit 169451e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb
@@ -120,7 +120,7 @@ def id3_discrete(data, attributes, attribute)
       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