1 ! Copyright (C) 2012 Doug Coleman.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors fry kernel locals machine-learning.transformer
4 sequences sets sorting sorting.extras ;
5 IN: machine-learning.label-encoder
7 TUPLE: label-encoder classes_ ;
9 : <label-encoder> ( -- le ) label-encoder new ; inline
11 M: label-encoder fit-y ( y transformer -- )
12 [ members natural-sort ] dip classes_<< ;
14 M: label-encoder transform-y ( y transformer -- y' )
15 classes_>> '[ _ bisect-left ] map ;
17 M: label-encoder inverse-transform-y ( y' transformer -- y )
18 classes_>> '[ _ nth ] map ;