steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

math.statistics: faster and simpler entropy word.

db4
John Benediktsson 2012-07-11 12:26:20 -07:00
parent f019ee8776
commit bdf02bfd2b
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
basis/math/statistics/statistics.factor
View File

@ -330,8 +330,7 @@ ALIAS: corr sample-corr
[ ?first ] keep [ max dup ] map nip ; [ ?first ] keep [ max dup ] map nip ;
: entropy ( seq -- n ) : entropy ( seq -- n )
dup members [ [ = ] curry count ] with map histogram values dup sum '[ _ / dup log * ] map-sum neg ;
dup sum v/n dup [ log ] map v* sum neg ;
: binary-entropy ( p -- h ) : binary-entropy ( p -- h )
[ dup log * ] [ 1 swap - dup log * ] bi + neg 2 log / ; [ dup log * ] [ 1 swap - dup log * ] bi + neg 2 log / ;