steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

machine-learning: use bisect-left (new name for search-sorted).

db4
John Benediktsson 2013-11-16 16:24:08 -08:00
parent 53592b9c7d
commit 3ad4602b1c
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extra/machine-learning/label-binarizer/label-binarizer.factor
View File

@ -1,7 +1,7 @@
! Copyright (C) 2012 Doug Coleman. ! Copyright (C) 2012 Doug Coleman.
! See http://factorcode.org/license.txt for BSD license. ! See http://factorcode.org/license.txt for BSD license.
USING: accessors arrays fry kernel machine-learning.transformer USING: accessors arrays fry kernel machine-learning.transformer
math.extras sequences sets sorting ; sequences sets sorting sorting.extras ;
IN: machine-learning.label-binarizer IN: machine-learning.label-binarizer
TUPLE: label-binarizer classes_ ; TUPLE: label-binarizer classes_ ;
@ -14,7 +14,7 @@ M: label-binarizer fit-y
M: label-binarizer transform-y M: label-binarizer transform-y
classes_>> dup length '[ classes_>> dup length '[
_ search-sorted [ 1 ] dip _ 0 <array> [ set-nth ] keep _ bisect-left [ 1 ] dip _ 0 <array> [ set-nth ] keep
] map ; ] map ;
M: label-binarizer inverse-transform-y M: label-binarizer inverse-transform-y

4
extra/machine-learning/label-encoder/label-encoder.factor
View File

@ -1,7 +1,7 @@
! Copyright (C) 2012 Doug Coleman. ! Copyright (C) 2012 Doug Coleman.
! See http://factorcode.org/license.txt for BSD license. ! See http://factorcode.org/license.txt for BSD license.
USING: accessors fry kernel locals machine-learning.transformer USING: accessors fry kernel locals machine-learning.transformer
math.extras sequences sets sorting ; sequences sets sorting sorting.extras ;
IN: machine-learning.label-encoder IN: machine-learning.label-encoder
TUPLE: label-encoder classes_ ; TUPLE: label-encoder classes_ ;
@ -12,7 +12,7 @@ M: label-encoder fit-y ( y transformer -- )
[ members natural-sort ] dip classes_<< ; [ members natural-sort ] dip classes_<< ;
M: label-encoder transform-y ( y transformer -- y' ) M: label-encoder transform-y ( y transformer -- y' )
classes_>> '[ _ search-sorted ] map ; classes_>> '[ _ bisect-left ] map ;
M: label-encoder inverse-transform-y ( y' transformer -- y ) M: label-encoder inverse-transform-y ( y' transformer -- y )
classes_>> '[ _ nth ] map ; classes_>> '[ _ nth ] map ;