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

De-sum 2

Browse Source
This commit is contained in:
Brian Underwood
2017-05-09 16:12:38 -04:00
committed by GitHub
parent 846448acd6
commit 92b9a86699
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