extra/machine-learning/label-binarizer/label-binarizer-tests.factor

   1 ! Copyright (C) 2012 Doug Coleman.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: kernel machine-learning.label-binarizer
   4 machine-learning.transformer tools.test ;
   5 IN: machine-learning.label-binarizer.tests
   6
   7 { { { 1 0 0 0 } { 0 0 0 1 } } } [
   8     <label-binarizer> { 1 2 6 4 2 } over fit-y
   9     { 1 6 } swap transform-y
  10 ] unit-test
  11
  12 { { 1 6 } } [
  13     <label-binarizer> { 1 2 6 4 2 } over fit-y
  14     { 1 6 } over transform-y swap inverse-transform-y
  15 ] unit-test