De-sum 2

igrigorik · May 9, 2017 · 92b9a86 · 92b9a86
1 parent 846448a
commit 92b9a86
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