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 #35 from cheerfulstoic/patch-3

Browse Source 
De-sum 2
This commit is contained in:
Ilya Grigorik
2018-04-06 11:56:47 -07:00
committed by GitHub
parent 5ba86e07e0 5be22ab1f6
commit 2486d0df99
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/core_extensions/array.rb
View File

@@ -2,7 +2,7 @@ class Array
def entropy def entropy
each_with_object(Hash.new(0)) do |i, result| each_with_object(Hash.new(0)) do |i, result|
result[i] += 1 result[i] += 1
end.values.sum do |count| end.values.inject(0, :+) do |count|
percentage = count.to_f / length percentage = count.to_f / length
-percentage * Math.log2(percentage) -percentage * Math.log2(percentage)