]> gitweb.factorcode.org Git - factor.git/commitdiff
machine-learning: use bisect-left (new name for search-sorted).
authorJohn Benediktsson <mrjbq7@gmail.com>
Sun, 17 Nov 2013 00:24:08 +0000 (16:24 -0800)
committerJohn Benediktsson <mrjbq7@gmail.com>
Sun, 17 Nov 2013 00:24:08 +0000 (16:24 -0800)
extra/machine-learning/label-binarizer/label-binarizer.factor
extra/machine-learning/label-encoder/label-encoder.factor

index 60fcfd6d43374b20bb9529170a3c70036937bf36..a233c32540424761c6102b030e525f765a2f386a 100644 (file)
@@ -1,7 +1,7 @@
 ! Copyright (C) 2012 Doug Coleman.
 ! See http://factorcode.org/license.txt for BSD license.
 USING: accessors arrays fry kernel machine-learning.transformer
-math.extras sequences sets sorting ;
+sequences sets sorting sorting.extras ;
 IN: machine-learning.label-binarizer
 
 TUPLE: label-binarizer classes_ ;
@@ -14,7 +14,7 @@ M: label-binarizer fit-y
 
 M: label-binarizer transform-y
     classes_>> dup length '[
-        _ search-sorted [ 1 ] dip _ 0 <array> [ set-nth ] keep
+        _ bisect-left [ 1 ] dip _ 0 <array> [ set-nth ] keep
     ] map ;
 
 M: label-binarizer inverse-transform-y
index 35f1f6ee15f518cc7f51e8ee09f6d2e587dd30f8..5c605a97c05fef6092bac9f7520f86c8686e4de3 100644 (file)
@@ -1,7 +1,7 @@
 ! Copyright (C) 2012 Doug Coleman.
 ! See http://factorcode.org/license.txt for BSD license.
 USING: accessors fry kernel locals machine-learning.transformer
-math.extras sequences sets sorting ;
+sequences sets sorting sorting.extras ;
 IN: machine-learning.label-encoder
 
 TUPLE: label-encoder classes_ ;
@@ -12,7 +12,7 @@ M: label-encoder fit-y ( y transformer -- )
     [ members natural-sort ] dip classes_<< ;
 
 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 )
     classes_>> '[ _ nth ] map ;