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

Fix sum error

Browse Source
This commit is contained in:
Sam Oliver
2021-09-17 13:00:14 +02:00
parent aad3b61d10
commit a58ec254f6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/decisiontree/id3_tree.rb
View File

@@ -127,12 +127,12 @@ 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.inject(0, :+) do |val| remainder = values.sort.inject(0) do |sum, 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
((classification.size.to_f / data.size) * classification.entropy) sum + ((classification.size.to_f / data.size) * classification.entropy)
end end
[data.classification.entropy - remainder, index] [data.classification.entropy - remainder, index]