average-gain ( dataset idx -- gain )

Vocabulary

machine-learning.decision-trees

Inputs

dataset	an object
idx	an object

Outputs

gain

an object

Definition

USING: accessors assocs math sequences ;

IN: machine-learning.decision-trees

:: average-gain ( dataset idx -- gain )
    dataset targets>> :> targets dataset features>> :> features
    features targets zip
    :> features-targets features-targets idx
    subsets-weighted-entropy
    :> weighted targets entropy2 weighted features length / - ;