highest-gain-index ( dataset -- idx )

Vocabulary

machine-learning.decision-trees

Inputs

dataset

an object

Outputs

idx

an object

Definition

USING: accessors kernel sequences sequences.extras ;

IN: machine-learning.decision-trees

: highest-gain-index ( dataset -- idx )
    dup feature-names>> length <iota> [ average-gain ] with map
    arg-max ;