label-binarizer

Vocabulary

machine-learning.label-binarizer

Definition

IN: machine-learning.label-binarizer

TUPLE: label-binarizer classes_ ;

Methods

USING: accessors kernel machine-learning.label-binarizer
machine-learning.transformer sets sorting ;

M: label-binarizer fit-y [ members sort ] dip classes_<< ;


USING: accessors kernel machine-learning.label-binarizer
machine-learning.transformer sequences ;

M: label-binarizer inverse-transform-y
    classes_>> [ [ 1 swap index ] ] dip [ nth ] curry compose
    map ;


USING: accessors arrays kernel machine-learning.label-binarizer
machine-learning.transformer sequences sorting.extras ;

M: label-binarizer transform-y
    classes_>> dup length
    [ [ bisect-left [ 1 ] dip ] curry ] dip
    [ 0 <array> [ set-nth ] keep ] curry compose map ;