dkam/decisiontree
1
0
Fork 0
mirror of https://github.com/dkam/decisiontree.git synced 2025-12-28 07:04:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #34 from cheerfulstoic/patch-2

Browse Source 
De-sum
This commit is contained in:
Ilya Grigorik
2017-05-09 16:17:02 -07:00
committed by GitHub
parent 846448acd6 92b9a86699
commit 169451e855
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/decisiontree/id3_tree.rb
View File

@@ -120,7 +120,7 @@ module DecisionTree
index = attributes.index(attribute) index = attributes.index(attribute)
values = data.map { |row| row[index] }.uniq 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| classification = data.each_with_object([]) do |row, result|
result << row.last if row[index] == val result << row.last if row[index] == val
end end