subsets-weighted-entropy ( data-target idx -- seq )


Vocabulary
machine-learning.decision-trees

Inputs
data-targetan object
idxan object


Outputs
seqan object


Definition
USING: assocs kernel math sequences ;

IN: machine-learning.decision-trees

: subsets-weighted-entropy ( data-target idx -- seq )
[ [ first ] ] dip [ swap nth ] curry compose group-by-sorted
[ [ second ] map ] assoc-map values
[ [ entropy2 ] [ length ] bi * ] map-sum ; inline